Want to Change to bf16 from fp32 of VAE dtype. #16

Open

Open

Want to Change to bf16 from fp32 of VAE dtype.#16

I've added model.to(torch.bfloat16) in nodes.py, but the dtype isn't changing. How can I make the change?

Metadata

Assignees

No one assigned

Labels

No labels

No labels

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests