Spaces:

rafaaa2105
/

diffusers-collection

Running on Zero

rafaaa2105 commited on Apr 12

Commit

cb8f4ad

•

1 Parent(s): bdf96ee

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -32,7 +32,7 @@ for model_name in model_list:
             add_watermarker=False,
         )
-        models[model_name].to(device)
     except Exception as e:
         print(f"Error loading model {model_name}: {e}")

             add_watermarker=False,
         )
+        models[model_name].to("cuda")
     except Exception as e:
         print(f"Error loading model {model_name}: {e}")