Spaces:
Running
on
Zero
Running
on
Zero
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 |
|