Lifeinhockey commited on
Commit
2db29d4
·
verified ·
1 Parent(s): a854895

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -83,7 +83,7 @@ def infer(
83
  prompt_embeds = long_prompt_encoder(prompt, pipe.tokenizer, pipe.text_encoder)
84
  negative_prompt_embeds = long_prompt_encoder(negative_prompt, pipe.tokenizer, pipe.text_encoder)
85
  prompt_embeds, negative_prompt_embeds = align_embeddings(prompt_embeds, negative_prompt_embeds)
86
- pipe.fuse_lora(lora_scale=lora_scale) # Коэфф. добавления lora
87
 
88
  params = {
89
  'prompt_embeds': prompt_embeds,
@@ -93,7 +93,6 @@ def infer(
93
  'width': width,
94
  'height': height,
95
  'generator': generator,
96
- 'lora_scale': lora_scale,
97
  }
98
 
99
  return pipe(**params).images[0]
 
83
  prompt_embeds = long_prompt_encoder(prompt, pipe.tokenizer, pipe.text_encoder)
84
  negative_prompt_embeds = long_prompt_encoder(negative_prompt, pipe.tokenizer, pipe.text_encoder)
85
  prompt_embeds, negative_prompt_embeds = align_embeddings(prompt_embeds, negative_prompt_embeds)
86
+ pipe.fuse_lora(lora_scale=0.01) # Коэфф. добавления lora
87
 
88
  params = {
89
  'prompt_embeds': prompt_embeds,
 
93
  'width': width,
94
  'height': height,
95
  'generator': generator,
 
96
  }
97
 
98
  return pipe(**params).images[0]