Spaces:

ehristoforu
/

Proteus-V0.3

Paused

ehristoforu commited on Jan 18

Commit

a0a6e09

•

1 Parent(s): 6e7da15

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -32,7 +32,7 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 if torch.cuda.is_available():
     pipe = StableDiffusionXLPipeline.from_pretrained(
         "dataautogpt3/ProteusV0.1",
-        use_safetensors=True,
     )
     if ENABLE_CPU_OFFLOAD:
         pipe.enable_model_cpu_offload()

 if torch.cuda.is_available():
     pipe = StableDiffusionXLPipeline.from_pretrained(
         "dataautogpt3/ProteusV0.1",
+        use_safetensors=False,
     )
     if ENABLE_CPU_OFFLOAD:
         pipe.enable_model_cpu_offload()