Update gradio_sd3.py

gradio_sd3.py

@@ -23,7 +23,7 @@ example_path = os.path.join(os.path.dirname(__file__), 'examples')
 fitdit_repo = "BoyuanJiang/FitDiT"
 repo_path = snapshot_download(repo_id=fitdit_repo)
 
-weight_dtype = torch.bfloat4
+weight_dtype = torch.bfloat16
 device = "cpu"
 transformer_garm = SD3Transformer2DModel_Garm.from_pretrained(os.path.join(repo_path, "transformer_garm"), torch_dtype=weight_dtype)
 transformer_vton = SD3Transformer2DModel_Vton.from_pretrained(os.path.join(repo_path, "transformer_vton"), torch_dtype=weight_dtype)