Skip to content

Diffusion training crashes with split torch_dtype/compute_dtype when the FSDP mesh is size 1 #3587

Description

@zyzhou5

Issue: #3578

Fix PR: #3589

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions