Update app.py
Browse files
app.py
CHANGED
@@ -31,7 +31,8 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
|
|
31 |
pipe.to('cuda')
|
32 |
|
33 |
#pipe = DiffusionPipeline.from_pretrained(base_model_id, torch_dtype=torch.float16, variant="fp16").to("cuda")
|
34 |
-
pipe.load_lora_weights(repo_name, ckpt_name)
|
|
|
35 |
pipe.fuse_lora()
|
36 |
# Ensure ddim scheduler timestep spacing set as trailing !!!
|
37 |
pipe.scheduler = TCDScheduler.from_config(pipe.scheduler.config)
|
|
|
31 |
pipe.to('cuda')
|
32 |
|
33 |
#pipe = DiffusionPipeline.from_pretrained(base_model_id, torch_dtype=torch.float16, variant="fp16").to("cuda")
|
34 |
+
#pipe.load_lora_weights(repo_name, ckpt_name)
|
35 |
+
pipe.load_lora_weights(repo_name, weight_name=ckpt_name)
|
36 |
pipe.fuse_lora()
|
37 |
# Ensure ddim scheduler timestep spacing set as trailing !!!
|
38 |
pipe.scheduler = TCDScheduler.from_config(pipe.scheduler.config)
|