vtp-t5-mmdit/run.sh at main · LLSean/vtp-t5-mmdit · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
python3 -m mmdit.scripts.prepare_edges2shoes_t2i --max_samples 32

python3 -m mmdit.scripts.cache_vtp_images \
  --config configs/t2i_vtp_256_smoke.yaml \
  --meta_jsonl data/edges2shoes_t2i/train_images.jsonl \
  --out_dir cache/edges2shoes_t2i_latents

python3 -m mmdit.scripts.cache_vtp_images \
  --config configs/t2i_vtp_256_flow_matching_anime_buckets.yaml \
  --meta_jsonl data/danbooru_surtr_solo_100k_t2i/train_images.jsonl \
  --out_dir cache/danbooru_surtr_solo_100k_t2i_latents_buckets


python3 -m mmdit.scripts.cache_vtp_images \
  --config configs/t2i_vtp_256_flow_matching_anime_buckets_big.yaml \
  --meta_jsonl data/danbooru_surtr_solo_100k_t2i/train_images.jsonl \
  --out_dir cache/danbooru_surtr_solo_100k_t2i_latents

# small dataset
python3 -m mmdit.scripts.cache_vtp_images \
  --config configs/t2i_vtp_256_flow_matching_anime_buckets.yaml \
  --meta_jsonl data/danbooru_surtr_solo_100k_t2i_small/train_images.jsonl \
  --out_dir cache/danbooru_surtr_solo_100k_t2i_latents_small

python3 -m mmdit.scripts.estimate_latent_scale \
  --cache_dir cache/danbooru_surtr_solo_100k_t2i_latents_small \
  --target_std 1.0 \
  --max_files 0 \
  --seed 42 \
  --print_yaml


# normal dataset
python3 -m mmdit.scripts.estimate_latent_scale \
  --cache_dir cache/danbooru_surtr_solo_100k_t2i_latents \
  --target_std 1.0 \
  --max_files 0 \
  --seed 42 \
  --print_yaml
torchrun --master_port 29501 --nproc_per_node=1 -m mmdit.scripts.train_t2v   --config configs/t2i_vtp_256_flow_matching_anime.yaml   --train_cache_dir cache/danbooru_surtr_solo_100k_t2i_latents


python3 -m mmdit.scripts.train_t2v   --config configs/t2i_vtp_256_flow_matching_anime_buckets.yaml   --train_cache_dir cache/danbooru_surtr_solo_100k_t2i_latents --resume


python3 -m mmdit.scripts.train_t2v   --config configs/t2i_vtp_256_flow_matching_anime_buckets_big.yaml   --train_cache_dir cache/danbooru_surtr_solo_100k_t2i_latents --resume