put model to cpu first, then gpu later

app.py

@@ -54,7 +54,7 @@ adaface = AdaFaceWrapper(pipeline_name="text2img", base_model_path=base_model_pa
                          adaface_encoder_cfg_scales=args.adaface_encoder_cfg_scales,
                          enabled_encoders=args.enabled_encoders,
                          unet_types=None, extra_unet_dirpaths=args.extra_unet_dirpaths, 
-                         unet_weights=args.unet_weights, device=device)
+                         unet_weights=args.unet_weights, device='cpu')
 
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     if randomize_seed:
@@ -80,6 +80,8 @@ def generate_image(image_paths, guidance_scale, do_neg_id_prompt_weight, perturb
 
     global adaface
 
+    adaface.to(device)
+
     if image_paths is None or len(image_paths) == 0:
         raise gr.Error(f"Cannot find any input face image! Please upload a face image.")
     
@@ -127,7 +129,7 @@ def check_prompt_and_model_type(prompt, model_style_type):
     if model_style_type != args.model_style_type:
         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=device)
+                                 adaface_ckpt_paths=args.adaface_ckpt_path, device='cpu')
         # Update base model type.
         args.model_style_type = model_style_type
 
@@ -142,6 +144,10 @@ title = r"""
 description = r"""
 <b>Official demo</b> for our working paper <b>AdaFace: A Versatile Face Encoder for Zero-Shot Diffusion Model Personalization</b>.<br>
 
+❗️**What's New**❗️
+- Support switching between two model styles: **Realistic** and **Anime**.
+- If you just changed the model style, the first image/video generation will take extra 20~30 seconds for loading new model weight.
+
 ❗️**Tips**❗️
 1. Upload one or more images of a person. If multiple faces are detected, we use the largest one. 
 2. Check "Enhance Face" to highlight fine facial features.