This commit is contained in:
wangshuai6
2025-04-11 16:51:33 +08:00
parent 609cf377cb
commit ed3c466d08
2 changed files with 166 additions and 4 deletions

View File

@@ -38,13 +38,12 @@ pip install -r requirements.txt
# for inference
python main.py predict -c configs/repa_improved_ddt_xlen22de6_256.yaml --ckpt_path=XXX.ckpt
```
```bash
# extract image latent (optional)
python3 tools/cache_imlatent4.py
```
```bash
# for training
# extract image latent (optional)
python3 tools/cache_imlatent4.py
# train
python main.py fit -c configs/repa_improved_ddt_xlen22de6_256.yaml
```