RealVis_v5.0_BF16_E

Paused

ford442 commited on Dec 2, 2024

Commit

42336a6

1 Parent(s): de1140c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -117,7 +117,7 @@ def load_and_prepare_model(model_id):
         vae=vae,
     )
     pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
-    pipe.to(torch.bfloat16)
     pipe.to('cuda')
     return pipe

         vae=vae,
     )
     pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
+    #pipe.to(torch.bfloat16)
     pipe.to('cuda')
     return pipe