DarianT commited on
Commit
70854e0
·
1 Parent(s): f1c9730

Update LoRA path

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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