EquiFashionModel

README.md

@@ -105,7 +105,7 @@ from cldm.model import create_model, load_state_dict
 import torch
 
 # Download checkpoint
-ckpt = hf_hub_download("NguyenDinhHieu/EquiFashionModel", filename="hfd_100epochs.ckpt")
+ckpt = hf_hub_download("NguyenDinhHieu/EquiFashionModel", filename="eqf_final.ckpt")
 
 # Load model
 model = create_model("utils/configs/cldm_v2.yaml").to("cuda")
@@ -133,7 +133,7 @@ If you use this model or dataset, please cite:
 
 | File | Description |
 |------|--------------|
-| `hfd_100epochs.ckpt` | Main diffusion model checkpoint |
+| `eqf_final.ckpt` | Main Hybrid GAN–Diffusion model checkpoint |
 | `body_pose_model.pth`, `hand_pose_model.pth` | OpenPose keypoint weights |
 | `open_clip_pytorch_model.bin` | Pretrained OpenCLIP text encoder |
 | `app.py` | Gradio demo UI |