fffiloni commited on
Commit
3b3533f
·
verified ·
1 Parent(s): 0c95e8a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -88,6 +88,8 @@ def load_b_lora_to_unet(pipe, content_lora_model_id: str = '', style_lora_model_
88
 
89
 
90
  def load_b_loras(content_b_lora, style_b_lora):
 
 
91
 
92
  if content_b_lora != "" and content_b_lora is not None:
93
  # Get instance_prompt a.k.a trigger word
@@ -141,8 +143,6 @@ def main(content_b_lora, style_b_lora, prompt, negative_prompt, seed, randomize_
141
  width = width,
142
  height = height,
143
  ).images[0]
144
-
145
- pipeline.unload_lora_weights()
146
 
147
  return image, seed
148
 
 
88
 
89
 
90
  def load_b_loras(content_b_lora, style_b_lora):
91
+
92
+ pipeline.unload_lora_weights()
93
 
94
  if content_b_lora != "" and content_b_lora is not None:
95
  # Get instance_prompt a.k.a trigger word
 
143
  width = width,
144
  height = height,
145
  ).images[0]
 
 
146
 
147
  return image, seed
148