update to safetensors

app.py

@@ -14,7 +14,7 @@ from utils import align_face
 base_model_path = "stabilityai/stable-diffusion-xl-base-1.0"
 image_encoder_path = snapshot_download("orpatashnik/NestedAttentionEncoder", allow_patterns=["image_encoder/**"])
 image_encoder_path = os.path.join(image_encoder_path, "image_encoder")
-personalization_ckpt = hf_hub_download("orpatashnik/NestedAttentionEncoder", "personalization_encoder/pytorch_model.bin")
+personalization_ckpt = hf_hub_download("orpatashnik/NestedAttentionEncoder", "personalization_encoder/pytorch_model.safetensors")
 device = "cuda"
 
 # Special token settings
@@ -46,6 +46,7 @@ guidance_scale = 5.0
 # ----------------------
 # Inference function with alignment
 # ----------------------
+@spaces.GPU
 def generate_images(img1, img2, img3, prompt, w, num_samples, seed):
     # Collect non-empty reference images
     refs = [img for img in (img1, img2, img3) if img is not None]

requirements.txt

@@ -8,4 +8,5 @@ einops
 scipy
 accelerate
 insightface
-onnxruntime
+onnxruntime
+safetensors