Update app.py

app.py

@@ -26,7 +26,8 @@ from models.eva_clip.utils_qformer import resize_numpy_image_long
 os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True"
 os.environ["ZERO_GPU_PATCH_TORCH_DEVICE"] = "True"
 device = "cuda" if torch.cuda.is_available() else "cpu"
-
+from accelerate import Accelerator
+accelerator=Accelerator()
 hf_hub_download(repo_id="ai-forever/Real-ESRGAN", filename="RealESRGAN_x4.pth", local_dir="model_real_esran")
 snapshot_download(repo_id="AlexWortega/RIFE", local_dir="model_rife")
 snapshot_download(repo_id="BestWishYsh/ConsisID-preview", local_dir="BestWishYsh/ConsisID-preview")
@@ -87,7 +88,7 @@ face_helper.face_det.to(device)
 face_helper.face_parse.to(device)
 transformer.to(device, dtype=dtype)
 
-pipe = ConsisIDPipeline.from_pretrained(model_path, transformer=transformer, scheduler=scheduler, torch_dtype=dtype)
+pipe = accelerator.prepare(ConsisIDPipeline.from_pretrained(model_path, transformer=transformer, scheduler=scheduler, torch_dtype=dtype))
 # If you're using with lora, add this code
 if lora_path:
     pipe.load_lora_weights(lora_path, weight_name="pytorch_lora_weights.safetensors", adapter_name="test_1")