Update README.md

README.md

@@ -5,9 +5,9 @@ tags: []
 
 # Scaling Down Text Encoders of Text-to-Image Diffusion Models
 
-Official Repository of the paper: *[Scaling Down Text Encoders of Text-to-Image Diffusion Models](https://github.com/LifuWang-66/ScalingDownTextEncoder/tree/main)*.
+Official Repository of the paper: *[Scaling Down Text Encoders of Text-to-Image Diffusion Models](https://github.com/LifuWang-66/DistillT5)*.
 
-Project Page: https://lifuwang-66.github.io/ScalingDownTE/
+Project Page: https://github.com/LifuWang-66/DistillT5.git
 
 ## Model Descriptions:
 T5-Base distilled from [T5-XXL](https://huggingface.co/google/flan-t5-xxl) using [Flux](https://huggingface.co/runwayml/stable-diffusion-v1-5). 
@@ -22,8 +22,8 @@ It is 50 times smaller and retains most capability of T5-XXL.
 ## Usage:
 1. Setup the environment:
 ```
-git clone https://github.com/LifuWang-66/ScalingDownTextEncoder.git
-cd ScalingDownTextEncoder
+git clone https://github.com/LifuWang-66/DistillT5.git
+cd DistillT5
 conda create -n scaling python=3.12
 conda activate scaling
 pip install -r requirements.txt
@@ -41,7 +41,7 @@ from diffusers import FluxPipeline
 
 
 pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.float16)
-text_encoder = T5EncoderWithProjection.from_pretrained('lwang717/ScalingDownTextEncoder', torch_dtype=torch.float16)
+text_encoder = T5EncoderWithProjection.from_pretrained('LifuWang/DistillT5', torch_dtype=torch.float16)
 pipe.text_encoder_2 = text_encoder
 pipe = pipe.to('cuda')