Spaces:
Runtime error
Runtime error
yupeng.zhou
commited on
Commit
·
2546e87
1
Parent(s):
73d8370
fix
Browse files
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}")
|