ysharma HF staff commited on
Commit
797e0cc
·
1 Parent(s): fc4c790

update example weight

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -26,14 +26,14 @@ def monkeypatching(alpha, in_prompt, wt): #, prompt, pipe): finetuned_lora_weigh
26
  print(f"in_prompt is - {str(in_prompt)}")
27
  global counter
28
  if counter == 0 :
29
- if wt == "./lora_playgroundai_wt.pt" :
30
- monkeypatch_lora(pipe.unet, torch.load(wt)) #finetuned_lora_weights
31
- tune_lora_scale(pipe.unet, alpha) #1.00)
32
- counter +=1
33
- else:
34
- monkeypatch_lora(pipe.unet, torch.load("./output_example/lora_weight.pt")) #finetuned_lora_weights
35
- tune_lora_scale(pipe.unet, alpha) #1.00)
36
- counter +=1
37
  else :
38
  tune_lora_scale(pipe.unet, alpha) #1.00)
39
  prompt = "style of hclu, " + str(in_prompt) #"baby lion"
 
26
  print(f"in_prompt is - {str(in_prompt)}")
27
  global counter
28
  if counter == 0 :
29
+ #if wt == "./lora_playgroundai_wt.pt" :
30
+ monkeypatch_lora(pipe.unet, torch.load(wt)) #finetuned_lora_weights
31
+ tune_lora_scale(pipe.unet, alpha) #1.00)
32
+ counter +=1
33
+ #else:
34
+ #monkeypatch_lora(pipe.unet, torch.load("./output_example/lora_weight.pt")) #finetuned_lora_weights
35
+ #tune_lora_scale(pipe.unet, alpha) #1.00)
36
+ #counter +=1
37
  else :
38
  tune_lora_scale(pipe.unet, alpha) #1.00)
39
  prompt = "style of hclu, " + str(in_prompt) #"baby lion"