RealVis_v5.0_BF16_IP_B

Running on Zero

ford442 commited on Nov 26, 2024

Commit

b507387

1 Parent(s): d1d41ee

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -100,7 +100,7 @@ def load_and_prepare_model(model_id):
         add_watermarker=False,
     )
     pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
-    pipe.to.('cuda')
     if USE_TORCH_COMPILE:
         pipe.compile()

         add_watermarker=False,
     )
     pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
+    pipe.to('cuda')
     if USE_TORCH_COMPILE:
         pipe.compile()