yupeng.zhou commited on
Commit
ff212d9
1 Parent(s): d47f112
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -529,7 +529,7 @@ def process_generation(_sd_type,_model_type,_upload_images, _num_steps,style_nam
529
  # pipe.scheduler.set_timesteps(50)
530
  set_attention_processor(pipe.unet,id_length_,is_ipadapter = False)
531
  elif _model_type == "Photomaker":
532
- if _sd_type != "RealVision":
533
  pipe = pipe2.to(device)
534
  pipe.id_encoder.to(device)
535
  set_attention_processor(pipe.unet,id_length_,is_ipadapter = False)
@@ -732,7 +732,7 @@ with gr.Blocks(css=css) as demo:
732
 
733
  gr.Examples(
734
  examples=[
735
- [1,0.5,0.5,3,"a woman img, wearing a white T-shirt, blue loose hair",
736
  "bad anatomy, bad hands, missing fingers, extra fingers, three hands, three legs, bad arms, missing legs, missing arms, poorly drawn face, bad face, fused face, cloned face, three crus, fused feet, fused thigh, extra crus, ugly fingers, horn, cartoon, cg, 3d, unreal, animate, amputation, disconnected limbs",
737
  array2string(["wake up in the bed",
738
  "have breakfast",
 
529
  # pipe.scheduler.set_timesteps(50)
530
  set_attention_processor(pipe.unet,id_length_,is_ipadapter = False)
531
  elif _model_type == "Photomaker":
532
+ if _sd_type != "RealVision" and style_name != "(No style)":
533
  pipe = pipe2.to(device)
534
  pipe.id_encoder.to(device)
535
  set_attention_processor(pipe.unet,id_length_,is_ipadapter = False)
 
732
 
733
  gr.Examples(
734
  examples=[
735
+ [1,0.5,0.5,2,"a woman img, wearing a white T-shirt, blue loose hair",
736
  "bad anatomy, bad hands, missing fingers, extra fingers, three hands, three legs, bad arms, missing legs, missing arms, poorly drawn face, bad face, fused face, cloned face, three crus, fused feet, fused thigh, extra crus, ugly fingers, horn, cartoon, cg, 3d, unreal, animate, amputation, disconnected limbs",
737
  array2string(["wake up in the bed",
738
  "have breakfast",