update

embodied_gen/models/text_model.py

@@ -86,8 +86,7 @@ def build_text2img_ip_pipeline(
     ref_scale: float,
     device: str = "cuda",
 ) -> StableDiffusionXLPipelineIP:
-    if not os.path.exists(ckpt_dir):
-        download_kolors_weights(ckpt_dir)
+    download_kolors_weights(ckpt_dir)
 
     text_encoder = ChatGLMModel.from_pretrained(
         f"{ckpt_dir}/text_encoder", torch_dtype=torch.float16
@@ -140,8 +139,7 @@ def build_text2img_pipeline(
     ckpt_dir: str,
     device: str = "cuda",
 ) -> StableDiffusionXLPipeline:
-    if not os.path.exists(ckpt_dir):
-        download_kolors_weights(ckpt_dir)
+    download_kolors_weights(ckpt_dir)
 
     text_encoder = ChatGLMModel.from_pretrained(
         f"{ckpt_dir}/text_encoder", torch_dtype=torch.float16

embodied_gen/models/texture_model.py

@@ -42,8 +42,7 @@ def build_texture_gen_pipe(
     ip_adapt_scale: float = 0,
     device: str = "cuda",
 ) -> DiffusionPipeline:
-    if not os.path.exists(f"{base_ckpt_dir}/Kolors"):
-        download_kolors_weights(f"{base_ckpt_dir}/Kolors")
+    download_kolors_weights(f"{base_ckpt_dir}/Kolors")
         
     tokenizer = ChatGLMTokenizer.from_pretrained(
         f"{base_ckpt_dir}/Kolors/text_encoder"