internal_development

Runtime error

Menyu commited on May 11

Commit

9b99125

verified ·

1 Parent(s): bb34e02

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 = StableDiffusionXLPipeline.from_single_file(
-        "https://huggingface.co/bluepen5805/illustrious_pencil-XL/illustrious_pencil-XL-v2.0.0.safetensors",
         vae=vae,
         use_safetensors=True,
         torch_dtype=torch.float16,

 if torch.cuda.is_available():
     vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
     pipe = StableDiffusionXLPipeline.from_single_file(
+        "https://huggingface.co/bluepen5805/illustrious_pencil-XL/illustrious_pencil-XL-v4.0.0.safetensors",
         vae=vae,
         use_safetensors=True,
         torch_dtype=torch.float16,