flash-lora-araminta-k-styles

Runtime error

alvdansen commited on Jul 3, 2024

Commit

d1cf1e3

verified ·

1 Parent(s): caa404e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -31,7 +31,7 @@ model_id = "stabilityai/stable-diffusion-xl-base-1.0"
 pipe = DiffusionPipeline.from_pretrained(model_id, variant="fp16")
 pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
-pipe.load_lora_weights("jasperai/flash-sdxl", adapter_name="lora")
 pipe.to(device=DEVICE, dtype=torch.float16)

 pipe = DiffusionPipeline.from_pretrained(model_id, variant="fp16")
 pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
+pipe.load_lora_weights("alvdansen/flash-lora-araminta-k-styles", adapter_name="lora")
 pipe.to(device=DEVICE, dtype=torch.float16)