Fix device bug

app.py

@@ -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)
+        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')