Fixed optimizers parsing error in bash scripts (#61)

* replaced lambda statements with local functions to work with pickle in dataset.py and moved collate function outside of main and added its own class in cogvideox_image_to_video_lora.py

* fixed merge issue

* fixed row deletion merge

* fixed row deletion merge

* fixed comma-separated param issue in bash

README.md

@@ -78,7 +78,7 @@ Before starting the training, please check whether the dataset has been prepared
   ```bash
   LEARNING_RATES=("1e-4" "1e-3")
   LR_SCHEDULES=("cosine_with_restarts")
-  OPTIMIZERS=("adamw", "adam")
+  OPTIMIZERS=("adamw" "adam")
   MAX_TRAIN_STEPS=("3000")
   ```
 

README_zh.md

@@ -77,7 +77,7 @@ diffusers（该分支为 CogVideoX 的图像到视频添加了 LoRA 加载支持
   ```
   LEARNING_RATES=("1e-4" "1e-3")
   LR_SCHEDULES=("cosine_with_restarts")
-  OPTIMIZERS=("adamw", "adam")
+  OPTIMIZERS=("adamw" "adam")
   MAX_TRAIN_STEPS=("3000")
   ```
 

train_image_to_video_lora.sh

@@ -10,7 +10,7 @@ GPU_IDS="0"
 # Experiment with as many hyperparameters as you want!
 LEARNING_RATES=("1e-4" "1e-3")
 LR_SCHEDULES=("cosine_with_restarts")
-OPTIMIZERS=("adamw", "adam")
+OPTIMIZERS=("adamw" "adam")
 MAX_TRAIN_STEPS=("3000")
 
 # Single GPU uncompiled training

train_text_to_video_lora.sh

@@ -10,7 +10,7 @@ GPU_IDS="0"
 # Experiment with as many hyperparameters as you want!
 LEARNING_RATES=("1e-4" "1e-3")
 LR_SCHEDULES=("cosine_with_restarts")
-OPTIMIZERS=("adamw", "adam")
+OPTIMIZERS=("adamw" "adam")
 MAX_TRAIN_STEPS=("3000")
 
 # Single GPU uncompiled training