Spaces:

orpatashnik
/

NestedAttentionEncoder

Running on Zero

orpatashnik commited on 6 days ago

Commit

8eeff9c

1 Parent(s): ad42387

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -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.safetensors")
 device = "cuda"
 # Special token settings

 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/model.safetensors")
 device = "cuda"
 # Special token settings