RealVis_v5.0_BF16_IP

Running on Zero

ford442 commited on Dec 9, 2024

Commit

c63126b

1 Parent(s): 4592308

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -125,7 +125,7 @@ def load_and_prepare_model(model_id):
     #sched = DDIMScheduler.from_config(pipe.scheduler.config)
     pipe.scheduler=sched
     #pipe.to(device=device, dtype=torch.bfloat16)
-    pipe.to(device=device)
     return pipe
 # Preload and compile both models

     #sched = DDIMScheduler.from_config(pipe.scheduler.config)
     pipe.scheduler=sched
     #pipe.to(device=device, dtype=torch.bfloat16)
+    pipe.to('cuda')
     return pipe
 # Preload and compile both models