Spaces:

Menyu
/

NoobVPre06

Running on Zero

Menyu commited on Nov 8, 2024

Commit

6219619

verified ·

1 Parent(s): 874ecc6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -227,7 +227,7 @@ MAX_IMAGE_SIZE = 4096
 if torch.cuda.is_available():
     vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
     pipe = AutoPipelineForText2Image.from_pretrained(
-        "Yntec/noobai-xl-nai-xl-vpredtestversion-sdxl",
         torch_dtype=torch.float16,
         use_safetensors=True,
         add_watermarker=False

 if torch.cuda.is_available():
     vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
     pipe = AutoPipelineForText2Image.from_pretrained(
+        "Menyu/Vpred-checkpoints5000",
         torch_dtype=torch.float16,
         use_safetensors=True,
         add_watermarker=False