Spaces:

adaface-neurips
/

adaface-animate

Running on Zero

adaface-neurips commited on 8 days ago

Commit

2bf6b24

•

1 Parent(s): d2b3308

Fix device bug

Files changed (1) hide show

app.py CHANGED Viewed

@@ -223,7 +223,7 @@ def check_prompt_and_model_type(prompt, model_style_type):
  base_model_path = model_style_type2base_model_path[model_style_type]
  # If the base model type is changed, reload the model.
  if model_style_type != args.model_style_type:
- id_animator = load_model(model_style_type=model_style_type, device=device)
  adaface = AdaFaceWrapper(pipeline_name="text2img", base_model_path=base_model_path,
  adaface_encoder_types=args.adaface_encoder_types,
  adaface_ckpt_paths=[args.adaface_ckpt_path], device='cpu')

  base_model_path = model_style_type2base_model_path[model_style_type]
  # If the base model type is changed, reload the model.
  if model_style_type != args.model_style_type:
+ id_animator = load_model(model_style_type=model_style_type, device='cpu')
  adaface = AdaFaceWrapper(pipeline_name="text2img", base_model_path=base_model_path,
  adaface_encoder_types=args.adaface_encoder_types,
  adaface_ckpt_paths=[args.adaface_ckpt_path], device='cpu')