Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ dtype = torch.float16
|
|
17 |
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
18 |
|
19 |
repo = "fluently/Fluently-XL-Final"
|
20 |
-
pipe = StableDiffusionXLPipeline.from_pretrained(repo, torch_dtype=torch.float16)
|
21 |
pipe.load_lora_weights("KingNish/Better-Image-XL-Lora", weight_name="example-03.safetensors", adapter_name="lora")
|
22 |
pipe.set_adapters("lora")
|
23 |
pipe.to("cuda")
|
|
|
17 |
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
18 |
|
19 |
repo = "fluently/Fluently-XL-Final"
|
20 |
+
pipe = StableDiffusionXLPipeline.from_pretrained(repo, torch_dtype=torch.float16, vae=vae)
|
21 |
pipe.load_lora_weights("KingNish/Better-Image-XL-Lora", weight_name="example-03.safetensors", adapter_name="lora")
|
22 |
pipe.set_adapters("lora")
|
23 |
pipe.to("cuda")
|