Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -15,7 +15,7 @@ ENABLE_CPU_OFFLOAD = "1"
|
|
15 |
|
16 |
if torch.cuda.is_available():
|
17 |
pipe = AutoPipelineForText2Image.from_pretrained(
|
18 |
-
"
|
19 |
torch_dtype=torch.float16,
|
20 |
use_safetensors=True,
|
21 |
add_watermarker=False
|
|
|
15 |
|
16 |
if torch.cuda.is_available():
|
17 |
pipe = AutoPipelineForText2Image.from_pretrained(
|
18 |
+
"WhiteAiZ/noobaiXLNAIXL_epsilonPred075_diffusers",
|
19 |
torch_dtype=torch.float16,
|
20 |
use_safetensors=True,
|
21 |
add_watermarker=False
|