Lifeinhockey commited on
Commit
9757d1a
·
verified ·
1 Parent(s): 8b9a2dd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -7
app.py CHANGED
@@ -401,14 +401,18 @@ def infer(
401
 
402
  pipe_controlnet.scheduler = UniPCMultistepScheduler.from_config(pipe_controlnet.scheduler.config)
403
 
404
- prompt_embeds = long_prompt_encoder(prompt, pipe_controlnet.tokenizer, pipe_controlnet.text_encoder)
405
- negative_prompt_embeds = long_prompt_encoder(negative_prompt, pipe_controlnet.tokenizer, pipe_controlnet.text_encoder)
406
- prompt_embeds, negative_prompt_embeds = align_embeddings(prompt_embeds, negative_prompt_embeds)
 
 
 
 
407
  image = pipe_controlnet(
408
- # prompt=prompt,
409
- # negative_prompt=negative_prompt,
410
- prompt_embeds=prompt_embeds,
411
- negative_prompt_embeds=negative_prompt_embeds,
412
  image=cn_source_image,
413
  control_image=control_image,
414
  strength=strength_cn, # Коэфф. зашумления, чем больше, тем больше меняется результирующее изображение относитенльно исходного
 
401
 
402
  pipe_controlnet.scheduler = UniPCMultistepScheduler.from_config(pipe_controlnet.scheduler.config)
403
 
404
+ # Преобразуем изображения
405
+ cn_source_image = preprocess_image(cn_source_image, width, height)
406
+ control_image = preprocess_image(control_image, width, height)
407
+
408
+ # prompt_embeds = long_prompt_encoder(prompt, pipe_controlnet.tokenizer, pipe_controlnet.text_encoder)
409
+ # negative_prompt_embeds = long_prompt_encoder(negative_prompt, pipe_controlnet.tokenizer, pipe_controlnet.text_encoder)
410
+ # prompt_embeds, negative_prompt_embeds = align_embeddings(prompt_embeds, negative_prompt_embeds)
411
  image = pipe_controlnet(
412
+ prompt=prompt,
413
+ negative_prompt=negative_prompt,
414
+ # prompt_embeds=prompt_embeds,
415
+ # negative_prompt_embeds=negative_prompt_embeds,
416
  image=cn_source_image,
417
  control_image=control_image,
418
  strength=strength_cn, # Коэфф. зашумления, чем больше, тем больше меняется результирующее изображение относитенльно исходного