fix paddlemix ce (#3014)

PaddlePaddle · Jan 8, 2025 · a54ca5e · a54ca5e
1 parent cceb324
commit a54ca5e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/models/PaddleMIX/CE/ppdiffusers/class_conditional_image_generation_DiT/prepare.sh b/models/PaddleMIX/CE/ppdiffusers/class_conditional_image_generation_DiT/prepare.sh
@@ -4,9 +4,10 @@ pip install -r requirements.txt
 
 #bash ${root_path}/PaddleMIX/change_paddlenlp_version.sh
 
-
+mkdir data
 rm -rf fastdit_imagenet256_tiny/
 wget https://bj.bcebos.com/v1/paddlenlp/datasets/paddlemix/fastdit_features/fastdit_imagenet256_tiny.tar
 tar -xf fastdit_imagenet256_tiny.tar
 rm -rf fastdit_imagenet256_tiny.tar
+mv fastdit_imagenet256_tiny ./data/fastdit_imagenet256
 
diff --git a/models/PaddleMIX/CE/ppdiffusers/class_conditional_image_generation_DiT/start.sh b/models/PaddleMIX/CE/ppdiffusers/class_conditional_image_generation_DiT/start.sh
@@ -19,6 +19,7 @@ exit_code=0
 
 # 下载依赖和数据
 # bash prepare_new.sh
+bash prepare.sh
 
 echo "*******class_conditional_image_generation/DiT train begin***********"
 (sh 0_run_train_dit_trainer.sh) 2>&1 | tee ${log_dir}/class_conditional_image_generation_DiT_train.log