diffusers
8421c146 - Use parameters + buffers when deciding upscale_dtype (#9882)

Comment changes are shownComment changes are hidden
Commit
199 days ago
Use parameters + buffers when deciding upscale_dtype (#9882) Sometimes, the decoder might lack parameters and only buffers (e.g., this happens when we manually need to convert all the parameters to buffers — e.g. to avoid packing fp16 and fp32 parameters with FSDP)
Author
Parents
  • src/diffusers/models/autoencoders
    • File
      autoencoder_kl_temporal_decoder.py