NoobVPre065s

Running on Zero

Menyu commited on 24 days ago

Commit

19c3084

•

1 Parent(s): 9b82c69

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -225,7 +225,7 @@ MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 4096
 if torch.cuda.is_available():
-    vae = StableDiffusionXLPipeline.from_single_file("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
     pipe = StableDiffusionXLPipeline.from_single_file(
         "https://huggingface.co/Laxhar/noobai-XL-Vpred-0.6/blob/main/noobai-xl-vpred-v0.6.safetensors",
         vae=vae,

 MAX_IMAGE_SIZE = 4096
 if torch.cuda.is_available():
+    vae = StableDiffusionXLPipeline.from_single_file("https://huggingface.co/madebyollin/sdxl-vae-fp16-fix/blob/main/sdxl.vae.safetensors", torch_dtype=torch.float16)
     pipe = StableDiffusionXLPipeline.from_single_file(
         "https://huggingface.co/Laxhar/noobai-XL-Vpred-0.6/blob/main/noobai-xl-vpred-v0.6.safetensors",
         vae=vae,