Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -32,7 +32,7 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
|
32 |
if torch.cuda.is_available():
|
33 |
pipe = StableDiffusionXLPipeline.from_pretrained(
|
34 |
"dataautogpt3/ProteusV0.1",
|
35 |
-
use_safetensors=
|
36 |
)
|
37 |
if ENABLE_CPU_OFFLOAD:
|
38 |
pipe.enable_model_cpu_offload()
|
|
|
32 |
if torch.cuda.is_available():
|
33 |
pipe = StableDiffusionXLPipeline.from_pretrained(
|
34 |
"dataautogpt3/ProteusV0.1",
|
35 |
+
use_safetensors=False,
|
36 |
)
|
37 |
if ENABLE_CPU_OFFLOAD:
|
38 |
pipe.enable_model_cpu_offload()
|