yupeng.zhou commited on
Commit
2546e87
1 Parent(s): 73d8370
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -504,6 +504,7 @@ def process_generation(_sd_type,_model_type,_upload_images, _num_steps,style_nam
504
  set_attention_processor(pipe.unet,id_length_,is_ipadapter = False)
505
  elif _model_type == "Photomaker":
506
  pipe = pipe2
 
507
  set_attention_processor(pipe.unet,id_length_,is_ipadapter = False)
508
  else:
509
  raise NotImplementedError("You should choice between original and Photomaker!",f"But you choice {_model_type}")
 
504
  set_attention_processor(pipe.unet,id_length_,is_ipadapter = False)
505
  elif _model_type == "Photomaker":
506
  pipe = pipe2
507
+ pipe2.id_encoder.to("cuda")
508
  set_attention_processor(pipe.unet,id_length_,is_ipadapter = False)
509
  else:
510
  raise NotImplementedError("You should choice between original and Photomaker!",f"But you choice {_model_type}")