adaface-neurips commited on
Commit
566ec8f
·
1 Parent(s): 578cfba

Update code of changing base model type

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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