RussianVibe

Runtime error

0x7o commited on Feb 18, 2024

Commit

f914c0e

verified ·

1 Parent(s): 2bb8b7a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -27,7 +27,7 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 if torch.cuda.is_available():
     vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
     pipe = DiffusionPipeline.from_pretrained(
-        "0x7o/RussianVibe-XL-v1.0",
         vae=vae,
         torch_dtype=torch.float16,
         use_safetensors=True,

 if torch.cuda.is_available():
     vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
     pipe = DiffusionPipeline.from_pretrained(
+        "stabilityai/stable-diffusion-xl-base-1.0",
         vae=vae,
         torch_dtype=torch.float16,
         use_safetensors=True,