Spaces:
Runtime error
Runtime error
Commit
·
566ec8f
1
Parent(s):
578cfba
Update code of changing base model type
Browse files
app.py
CHANGED
|
@@ -268,7 +268,9 @@ def check_prompt_and_model_type(prompt, model_style_type, progress=gr.Progress()
|
|
| 268 |
id_animator = load_model(model_style_type=model_style_type, device='cpu')
|
| 269 |
adaface = AdaFaceWrapper(pipeline_name="text2img", base_model_path=base_model_path,
|
| 270 |
adaface_encoder_types=args.adaface_encoder_types,
|
| 271 |
-
adaface_ckpt_paths=[args.adaface_ckpt_path], device='cpu'
|
|
|
|
|
|
|
| 272 |
# Update base model type.
|
| 273 |
args.model_style_type = model_style_type
|
| 274 |
|
|
|
|
| 268 |
id_animator = load_model(model_style_type=model_style_type, device='cpu')
|
| 269 |
adaface = AdaFaceWrapper(pipeline_name="text2img", base_model_path=base_model_path,
|
| 270 |
adaface_encoder_types=args.adaface_encoder_types,
|
| 271 |
+
adaface_ckpt_paths=[args.adaface_ckpt_path], device='cpu',
|
| 272 |
+
num_inference_steps=args.num_inference_steps,
|
| 273 |
+
is_on_hf_space=is_on_hf_space)
|
| 274 |
# Update base model type.
|
| 275 |
args.model_style_type = model_style_type
|
| 276 |
|