AdwolfCzar commited on
Commit
b726579
·
verified ·
1 Parent(s): 1bb4ac3

Upload configs/cache_smoothloopv3.sh with huggingface_hub

Browse files
Files changed (1) hide show
  1. configs/cache_smoothloopv3.sh +33 -0
configs/cache_smoothloopv3.sh ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env bash
2
+ # Cache de latentes e de text-encoder, SERIAL na mesma GPU.
3
+ # Paralelo entre as duas etapas piora ambas (medido no historico).
4
+ set -Eeuo pipefail
5
+ PIN=ede9ed9f769d605db32d37e32243d4c979f9eee3
6
+ : "${TRAINER_DIR:=/workspace/projects/musubi-tuner}"
7
+ : "${VENV:=/venv/main}"
8
+ : "${MODEL_DIR:=/workspace/models/MiniMax-H3}"
9
+ : "${DATASET_CONFIG:?caminho absoluto do TOML de CACHE}"
10
+ : "${TASK:=i2va}"
11
+ : "${CACHE_BATCH:=24}"
12
+ : "${CACHE_WORKERS:=16}"
13
+ [[ "$TASK" == i2va || "$TASK" == fl2va ]] || { echo "use i2va ou fl2va revisado" >&2; exit 2; }
14
+ source "$VENV/bin/activate"
15
+ cd "$TRAINER_DIR"
16
+ [[ "$(git rev-parse HEAD)" == "$PIN" ]] || { echo "pin do trainer errado" >&2; exit 2; }
17
+ [[ -s "$DATASET_CONFIG" ]] || { echo "config ausente" >&2; exit 2; }
18
+
19
+ echo "=== [1/2] latentes ==="
20
+ python src/musubi_tuner/minimax_h3_cache_latents.py \
21
+ --dataset_config "$DATASET_CONFIG" \
22
+ --vae "$MODEL_DIR/vae/minimax_h3_video_vae_fp16.safetensors" \
23
+ --device cuda --batch_size "$CACHE_BATCH" --num_workers "$CACHE_WORKERS" \
24
+ --skip_existing --keep_cache
25
+
26
+ echo "=== [2/2] text encoder ==="
27
+ python src/musubi_tuner/minimax_h3_cache_text_encoder_outputs.py \
28
+ --dataset_config "$DATASET_CONFIG" \
29
+ --text_encoder "$MODEL_DIR/text_encoders/qwen3vl_32b_minimax_h3_nvfp4_awq.safetensors" \
30
+ --text_encoder_quantization nvfp4_awq --task "$TASK" --cache_guidance_empty \
31
+ --device cuda --batch_size "$CACHE_BATCH" --num_workers "$CACHE_WORKERS" \
32
+ --skip_existing --keep_cache
33
+ echo "=== cache concluido ==="