diff --git a/README.md b/README.md index bd680b7..dde59cc 100644 --- a/README.md +++ b/README.md @@ -15,8 +15,8 @@ We offer both the full version and distilled models. For more information about | Name | Script | Inference Steps | HuggingFace repo | | --------------- | -------------------------------------------------- | --------------- | ---------------------- | | HiDream-I1-Full | [inference.py](./inference.py) | 50 | 🤗 [HiDream-I1-Full](https://huggingface.co/HiDream-ai/HiDream-I1-Full) | -| HiDream-I1-Dev | [inference.py](./inference_distilled.py) | 28 | 🤗 [HiDream-I1-Dev](https://huggingface.co/HiDream-ai/HiDream-I1-Dev) | -| HiDream-I1-Fast | [inference.py](./inference_distilled.py) | 16 | 🤗 [HiDream-I1-Fast](https://huggingface.co/HiDream-ai/HiDream-I1-Fast) | +| HiDream-I1-Dev | [inference.py](./inference.py) | 28 | 🤗 [HiDream-I1-Dev](https://huggingface.co/HiDream-ai/HiDream-I1-Dev) | +| HiDream-I1-Fast | [inference.py](./inference.py) | 16 | 🤗 [HiDream-I1-Fast](https://huggingface.co/HiDream-ai/HiDream-I1-Fast) | ## Quick Start @@ -93,4 +93,4 @@ python gradio_demo.py ## License -The code in this repository and the HiDream-I1 models are licensed under [MIT License](./LICENSE). \ No newline at end of file +The code in this repository and the HiDream-I1 models are licensed under [MIT License](./LICENSE).