Rename && Update Readme
This commit is contained in:
@@ -2,9 +2,9 @@
|
||||
|
||||
The default training commands for the different versions are as follows:
|
||||
|
||||
We can choose whether to use deep speed in CogVideoX-Arbitrary, which can save a lot of video memory.
|
||||
We can choose whether to use deep speed in CogVideoX-Fun, which can save a lot of video memory.
|
||||
|
||||
CogVideoX-Arbitrary without deepspeed:
|
||||
CogVideoX-Fun without deepspeed:
|
||||
```sh
|
||||
export MODEL_NAME="models/Diffusion_Transformer/CogVideoX-Fun-2b-InP"
|
||||
export DATASET_NAME="datasets/internal_datasets/"
|
||||
@@ -52,7 +52,7 @@ accelerate launch --mixed_precision="bf16" scripts/train.py \
|
||||
--trainable_modules "."
|
||||
```
|
||||
|
||||
CogVideoX-Arbitrary with deepspeed:
|
||||
CogVideoX-Fun with deepspeed:
|
||||
```sh
|
||||
export MODEL_NAME="models/Diffusion_Transformer/CogVideoX-Fun-2b-InP"
|
||||
export DATASET_NAME="datasets/internal_datasets/"
|
||||
|
||||
@@ -1,8 +1,8 @@
|
||||
## Lora Training Code
|
||||
|
||||
We can choose whether to use deep speed in CogVideoX-Arbitrary, which can save a lot of video memory.
|
||||
We can choose whether to use deep speed in CogVideoX-Fun, which can save a lot of video memory.
|
||||
|
||||
CogVideoX-Arbitrary without deepspeed:
|
||||
CogVideoX-Fun without deepspeed:
|
||||
|
||||
```sh
|
||||
export MODEL_NAME="models/Diffusion_Transformer/CogVideoX-Fun-2b-InP"
|
||||
|
||||
Reference in New Issue
Block a user