Spaces:
Running
on
Zero
Running
on
Zero
Update LoRA path
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
|
|
17 |
pipe = StableDiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch.float16).to(device)
|
18 |
pipe.scheduler = DDPMScheduler.from_pretrained(model_repo_id, subfolder="scheduler")
|
19 |
|
20 |
-
folder_of_lora_weights = "./
|
21 |
which_checkpoint = "checkpoint-31-6400"
|
22 |
lora_name = "pytorch_lora_weights.safetensors"
|
23 |
|
|
|
17 |
pipe = StableDiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch.float16).to(device)
|
18 |
pipe.scheduler = DDPMScheduler.from_pretrained(model_repo_id, subfolder="scheduler")
|
19 |
|
20 |
+
folder_of_lora_weights = "./ID-Booth_LoRA_weights"
|
21 |
which_checkpoint = "checkpoint-31-6400"
|
22 |
lora_name = "pytorch_lora_weights.safetensors"
|
23 |
|