modify README for better test guidance

open-mmlab · Sep 19, 2023 · 178a2e8 · 178a2e8
1 parent db8023d
commit 178a2e8
Show file tree

Hide file tree

Showing 7 changed files with 18 additions and 24 deletions.
diff --git a/configs/animatediff/README.md b/configs/animatediff/README.md
@@ -50,7 +50,7 @@ wget https://civitai.com/api/download/models/78775 -P DreamBooth_LoRA/ --content
 2. Modify the config file in `configs/animatediff/animatediff_ToonYou.py`
 
 ```python
-
+    models_path = {Your Checkpoints Path}
     motion_module_cfg=dict(
         path={Your MotionModule path}
     ),

diff --git a/configs/animatediff/README_zh-CN.md b/configs/animatediff/README_zh-CN.md
@@ -52,7 +52,7 @@ wget https://civitai.com/api/download/models/78775 -P DreamBooth_LoRA/ --content
 2. 修改 `configs/animatediff/animatediff_ToonYou.py` 配置文件中的权重路径
 
 ```python
-
+    models_path = {Your Checkpoints Path}
     motion_module_cfg=dict(
         path={Your MotionModule path}
     ),

diff --git a/configs/animatediff/animatediff_Lyriel.py b/configs/animatediff/animatediff_Lyriel.py
@@ -1,5 +1,6 @@
 # config for model
 stable_diffusion_v15_url = 'runwayml/stable-diffusion-v1-5'
+models_path = '/home/wangrunqi/glcc/AnimateDiff/models/'
 randomness = dict(
     seed=[
         10917152860782582783, 6399018107401806238, 15875751942533906793,
@@ -52,12 +53,9 @@
     scheduler=diffusion_scheduler,
     test_scheduler=diffusion_scheduler,
     data_preprocessor=dict(type='DataPreprocessor'),
-    motion_module_cfg=dict(
-        path='/home/wangrunqi/AnimateDiff/models/Motion_Module/mm_sd_v14.ckpt'
-    ),
+    motion_module_cfg=dict(path=models_path + 'Motion_Module/mm_sd_v14.ckpt'),
     dream_booth_lora_cfg=dict(
         type='ToonYou',
-        path='/home/wangrunqi/AnimateDiff/models/' +
-        'DreamBooth_LoRA/lyriel_v16.safetensors',
+        path=models_path + 'DreamBooth_LoRA/lyriel_v16.safetensors',
         steps=25,
         guidance_scale=7.5))
diff --git a/configs/animatediff/animatediff_MajicMix.py b/configs/animatediff/animatediff_MajicMix.py
@@ -1,5 +1,6 @@
 # config for model
 stable_diffusion_v15_url = 'runwayml/stable-diffusion-v1-5'
+models_path = '/home/wangrunqi/glcc/AnimateDiff/models/'
 randomness = dict(
     seed=[
         1572448948722921032, 1099474677988590681, 6488833139725635347,
@@ -52,12 +53,10 @@
     scheduler=diffusion_scheduler,
     test_scheduler=diffusion_scheduler,
     data_preprocessor=dict(type='DataPreprocessor'),
-    motion_module_cfg=dict(
-        path='/home/wangrunqi/AnimateDiff/models/Motion_Module/mm_sd_v14.ckpt'
-    ),
+    motion_module_cfg=dict(path=models_path + 'Motion_Module/mm_sd_v14.ckpt'),
     dream_booth_lora_cfg=dict(
         type='ToonYou',
-        path='/home/wangrunqi/AnimateDiff/models/' +
+        path=models_path +
         'DreamBooth_LoRA/majicmixRealistic_v5Preview.safetensors',
         steps=25,
         guidance_scale=7.5))
diff --git a/configs/animatediff/animatediff_RcnzCartoon.py b/configs/animatediff/animatediff_RcnzCartoon.py
@@ -1,5 +1,6 @@
 # config for model
 stable_diffusion_v15_url = 'runwayml/stable-diffusion-v1-5'
+models_path = '/home/wangrunqi/glcc/AnimateDiff/models/'
 randomness = dict(
     seed=[
         16931037867122267877, 2094308009433392066, 4292543217695451092,
@@ -52,12 +53,9 @@
     scheduler=diffusion_scheduler,
     test_scheduler=diffusion_scheduler,
     data_preprocessor=dict(type='DataPreprocessor'),
-    motion_module_cfg=dict(
-        path='/home/wangrunqi/AnimateDiff/models/Motion_Module/mm_sd_v14.ckpt'
-    ),
+    motion_module_cfg=dict(path=models_path + 'Motion_Module/mm_sd_v14.ckpt'),
     dream_booth_lora_cfg=dict(
         type='ToonYou',
-        path='/home/wangrunqi/AnimateDiff/models/' +
-        'DreamBooth_LoRA/rcnzCartoon3d_v10.safetensors',
+        path=models_path + 'DreamBooth_LoRA/rcnzCartoon3d_v10.safetensors',
         steps=25,
         guidance_scale=7.5))
diff --git a/configs/animatediff/animatediff_RealisticVision.py b/configs/animatediff/animatediff_RealisticVision.py
@@ -1,5 +1,6 @@
 # config for model
 stable_diffusion_v15_url = 'runwayml/stable-diffusion-v1-5'
+models_path = '/home/wangrunqi/glcc/AnimateDiff/models/'
 randomness = dict(
     seed=[
         5658137986800322009, 12099779162349365895, 10499524853910852697,
@@ -52,12 +53,10 @@
     scheduler=diffusion_scheduler,
     test_scheduler=diffusion_scheduler,
     data_preprocessor=dict(type='DataPreprocessor'),
-    motion_module_cfg=dict(
-        path='/home/wangrunqi/AnimateDiff/models/Motion_Module/mm_sd_v14.ckpt'
-    ),
+    motion_module_cfg=dict(path=models_path + 'Motion_Module/mm_sd_v14.ckpt'),
     dream_booth_lora_cfg=dict(
         type='ToonYou',
-        path='/home/wangrunqi/AnimateDiff/models/' +
+        path=models_path +
         'DreamBooth_LoRA/realisticVisionV20_v20.safetensors',
         steps=25,
         guidance_scale=7.5))
diff --git a/configs/animatediff/animatediff_RealisticVision_v2.py b/configs/animatediff/animatediff_RealisticVision_v2.py
@@ -1,5 +1,6 @@
 # config for model
 stable_diffusion_v15_url = 'runwayml/stable-diffusion-v1-5'
+models_path = '/home/wangrunqi/glcc/AnimateDiff/models/'
 randomness = dict(
     seed=[
         13100322578370451493, 14752961627088720670, 9329399085567825781,
@@ -53,12 +54,11 @@
     scheduler=diffusion_scheduler,
     test_scheduler=diffusion_scheduler,
     data_preprocessor=dict(type='DataPreprocessor'),
-    motion_module_cfg=dict(
-        path=
-        '/home/wangrunqi/AnimateDiff/models/Motion_Module/mm_sd_v15_v2.ckpt'),
+    motion_module_cfg=dict(path=models_path +
+                           'Motion_Module/mm_sd_v15_v2.ckpt'),
     dream_booth_lora_cfg=dict(
         type='ToonYou',
-        path='/home/wangrunqi/AnimateDiff/models/' +
+        path=models_path +
         'DreamBooth_LoRA/realisticVisionV20_v20.safetensors',
         steps=25,
         guidance_scale=7.5))