Update app.py

app.py

@@ -7,7 +7,6 @@ from insightface.app import FaceAnalysis
 from huggingface_hub import hf_hub_download
 import os
 import logging
-from safetensors.torch import load_file
 import time
 
 # Set up detailed logging
@@ -63,18 +62,11 @@ def download_file(repo_id, filename, local_dir, max_retries=3):
         return file_path
 
 # Define paths
-kolors_unet_path = "./unet"
 ip_adapter_path = "./"
-os.makedirs(kolors_unet_path, exist_ok=True)
 os.makedirs(ip_adapter_path, exist_ok=True)
 
-# Download weights with retries
+# Download IP-Adapter weights with retries
 logger.info("Starting weights download...")
-kolors_weights = download_file(
-    "Kwai-Kolors/Kolors",
-    "unet/diffusion_pytorch_model.fp16.safetensors",
-    kolors_unet_path
-)
 ip_adapter_weights = download_file(
     "Kwai-Kolors/Kolors-IP-Adapter-FaceID-Plus",
     "ipa-faceid-plus.bin",
@@ -110,16 +102,6 @@ except Exception as e:
     logger.error(f"Failed to load SDXL base model: {e}")
     raise
 
-# Load Kolors unet weights with strict=False
-logger.info(f"Loading Kolors unet weights from {kolors_weights}...")
-try:
-    state_dict = load_file(kolors_weights, device=device)
-    pipe.unet.load_state_dict(state_dict, strict=False)
-    logger.info("Kolors unet weights loaded successfully (with ignored mismatches).")
-except Exception as e:
-    logger.error(f"Failed to load Kolors unet weights: {e}")
-    raise
-
 # Load IP-Adapter
 logger.info(f"Loading IP-Adapter from {ip_adapter_weights}...")
 try: