Spaces:

Adarsh7700
/

duplicate-repo

Running

Adarsh7700 commited on 12 days ago

Commit

ff6df37

•

1 Parent(s): bbb6330

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -106,7 +106,7 @@ print('Loading diffusion model ...')
 pipeline = DiffusionPipeline.from_pretrained(
     "sudo-ai/zero123plus-v1.2",
     custom_pipeline="zero123plus",
-    torch_dtype=torch.float16,
 )
 pipeline.scheduler = EulerAncestralDiscreteScheduler.from_config(
     pipeline.scheduler.config, timestep_spacing='trailing'

 pipeline = DiffusionPipeline.from_pretrained(
     "sudo-ai/zero123plus-v1.2",
     custom_pipeline="zero123plus",
+    torch_dtype=torch.bfloat16,
 )
 pipeline.scheduler = EulerAncestralDiscreteScheduler.from_config(
     pipeline.scheduler.config, timestep_spacing='trailing'