Spaces:

softwareweaver
/

ImageGenPlus

Running on Zero

softwareweaver commited on Nov 7

Commit

f84d1c9

verified ·

1 Parent(s): b7848a2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -55,7 +55,8 @@ def infer(
     if model_name != current_model_id:
         pipe = DiffusionPipeline.from_pretrained(
             model_name,
-            torch_dtype=torch_dtype
         ).to(device)
         current_model_id = model_name

     if model_name != current_model_id:
         pipe = DiffusionPipeline.from_pretrained(
             model_name,
+            torch_dtype=torch_dtype,
+            use_auth_token=hf_token
         ).to(device)
         current_model_id = model_name