Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -227,7 +227,7 @@ MAX_IMAGE_SIZE = 4096
|
|
227 |
if torch.cuda.is_available():
|
228 |
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
229 |
pipe = AutoPipelineForText2Image.from_pretrained(
|
230 |
-
"
|
231 |
vae=vae,
|
232 |
torch_dtype=torch.float16,
|
233 |
use_safetensors=True,
|
|
|
227 |
if torch.cuda.is_available():
|
228 |
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
229 |
pipe = AutoPipelineForText2Image.from_pretrained(
|
230 |
+
"Menyu/noobaiXLNAIXL_vPred065SVersion",
|
231 |
vae=vae,
|
232 |
torch_dtype=torch.float16,
|
233 |
use_safetensors=True,
|