Spaces:
Paused
Paused
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 = "ID_Booth_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 |
|