zhicao/dreamzero main
SHELL
Verdict: BENIGN
Local reference
Suspicious dependency
Inferred
droid_training_wan22.sh
shell
| 1 |
#!/bin/bash |
|
| 2 |
# DreamZero DROID Training Script with Wan2.2-TI2V-5B backbone |
|
| 3 |
# |
|
| 4 |
# Usage: |
|
| 5 |
# bash scripts/train/droid_training_wan22.sh |
|
| 6 |
# |
|
| 7 |
# Prerequisites… |
|
| 18:13 |
…_FULL_ERROR=1 |
|
| 19 |
|
|
| 20 |
# Repo root: must be a directory that contains groot/ (so experiment.py can be found). |
|
| 21 |
# Beaker/weka uses /root/yejink/dreamzero; image has /root/dreamzero; else script location. |
References top-level /root path |
| 22 |
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)" |
Shell changes working directory |
| 23 |
SCRIPT_REPO_ROOT="$(cd "$SCRIPT_DIR/../.." && pwd)" |
|
| 24 |
if [ -n "$DREAMZERO_ROOT" ] && [ … |
|
| 79:125 |
…TA_ROOT" |
|
| 80 |
exit 1 |
|
| 81 |
fi |
|
| 82 |
|
|
| 83 |
# Use image Python 3.11 when available (Dockerfile installs dreamzero with python3.11 -m pip). |
Dockerfile |
| 84 |
# Absolute path so worker processes open the correct file even if their cwd differs… |
|
| 91:14 |
…HON_311" ]; then |
|
| 92 |
if [ -n "${FIX_NUMPY_IN_SCRIPT:-}" ]; then |
|
| 93 |
"$PYTHON_311" -m pip install "numpy==1.26.4" --force-reinstall -q 2>/dev/null || true |
Shell runs a variable command with args |
| 94 |
fi |
|
| 95 |
RUN_CMD=( "$PYTHON_311" -m torch.distributed.run --nproc_per_node "$NUM_GPUS" --standalone "$EXPERIMENT_PY" ) |
|
| 96 |
echo "Using image Python 3.11: $PYTHON_311" |
|
| 97 |
else |
|
| 98 |
RUN_CMD=( python3 -m torch.distributed.run --nproc_per_node "$NUM_GPUS" --standalone "$EXPERIMENT_PY" ) |
|
| 99 |
echo "Using: $(command -v python3)" |
|
| 100 |
fi |
|
| 101 |
cd "$DREAMZERO_ROOT" |
|
| 102 |
|
|
| 103 |
"${RUN_CMD[@]}" \ |
|
| 104 |
report_t… |
|
| 137:20 |
…N22_CKPT_DIR \ |
|
| 138 |
text_encoder_pretrained_path=$WAN22_CKPT_DIR/models_t5_umt5-xxl-enc-bf16.pth \ |
Checkpoint extension after model context |
| 139 |
image_encoder_pretrained_path=$IMAGE_ENCODER_DIR/models_clip_open-clip-xlm-roberta-large-vit-huge-14.pth \ |
|
| 140 |
vae_pretra… |
|
Micro-behaviors
| 18:13 |
…_FULL_ERROR=1 |
|
| 19 |
|
|
| 20 |
# Repo root: must be a directory that contains groot/ (so experiment.py can be found). |
|
| 21 |
# Beaker/weka uses /root/yejink/dreamzero; image has /root/dreamzero; else script location. |
|
| 22 |
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)" |
|
| 23 |
SCRIPT_REPO_ROOT="$(cd "$SCRIPT_DIR/../.." && pwd)" |
|
| 24 |
if [ -n "$DREAMZERO_ROOT" ] && [ … |
|
| 18:13 |
…_FULL_ERROR=1 |
|
| 19 |
|
|
| 20 |
# Repo root: must be a directory that contains groot/ (so experiment.py can be found). |
|
| 21 |
# Beaker/weka uses /root/yejink/dreamzero; image has /root/dreamzero; else script location. |
|
| 22 |
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)" |
|
| 23 |
SCRIPT_REPO_ROOT="$(cd "$SCRIPT_DIR/../.." && pwd)" |
|
| 24 |
if [ -n "$DREAMZERO_ROOT" ] && [ … |
|
| 137:20 |
…N22_CKPT_DIR \ |
|
| 138 |
text_encoder_pretrained_path=$WAN22_CKPT_DIR/models_t5_umt5-xxl-enc-bf16.pth \ |
|
| 139 |
image_encoder_pretrained_path=$IMAGE_ENCODER_DIR/models_clip_open-clip-xlm-roberta-large-vit-huge-14.pth \ |
|
| 140 |
vae_pretra… |
|
| 91:14 |
…HON_311" ]; then |
|
| 92 |
if [ -n "${FIX_NUMPY_IN_SCRIPT:-}" ]; then |
|
| 93 |
"$PYTHON_311" -m pip install "numpy==1.26.4" --force-reinstall -q 2>/dev/null || true |
|
| 94 |
fi |
|
| 95 |
RUN_CMD=( "$PYTHON_311" -m torch.distributed.run --nproc_per_node "$NUM_GPUS" --standalone "$EXPERIMENT_PY" ) |
|
| 96 |
echo "Using image Python 3.11: $PYTHON_311" |
|
| 97 |
else |
|
| 98 |
RUN_CMD=( python3 -m torch.distributed.run --nproc_per_node "$NUM_GPUS" --standalone "$EXPERIMENT_PY" ) |
|
| 99 |
echo "Using: $(command -v python3)" |
|
| 100 |
fi |
|
| 101 |
cd "$DREAMZERO_ROOT" |
|
| 102 |
|
|
| 103 |
"${RUN_CMD[@]}" \ |
|
| 104 |
report_t… |
|
Metadata
| 79:125 |
…TA_ROOT" |
|
| 80 |
exit 1 |
|
| 81 |
fi |
|
| 82 |
|
|
| 83 |
# Use image Python 3.11 when available (Dockerfile installs dreamzero with python3.11 -m pip). |
|
| 84 |
# Absolute path so worker processes open the correct file even if their cwd differs… |
|
-
0
#!/bin/bash
# DreamZero DROID Training Script with Wan2.2-TI2V-5B backbone
#
# Usage:
# bash scripts/train/droid_training_wan22.sh
#
# Prerequisites
| 1 |
#!/bin/bash |
|
| 2 |
# DreamZero DROID Training Script with Wan2.2-TI2V-5B backbone |
|
| 3 |
# |
|
| 4 |
# Usage: |
|
| 5 |
# bash scripts/train/droid_training_wan22.sh |
|
| 6 |
# |
|
| 7 |
# Prerequisites… |
|
| 71:21 |
…der not found. Downloading Wan2.1-I2V-14B-480P (for CLIP only)..." |
|
| 72 |
huggingface-cli download Wan-AI/Wan2.1-I2V-14B-480P --local-dir "$IMAGE_ENCO… |
|
| 24:43 |
…MZERO_ROOT/groot" ]; then |
|
| 25 |
: # keep existing and valid |
|
| 26 |
elif [ -d "/root/yejink/dreamzero/groot" ]; then |
|
| 27 |
DREAMZERO_ROOT=/root/yejink/dreamzero |
|
| 28 |
elif [ -d "/r… |
|
| 18:13 |
…_FULL_ERROR=1 |
|
| 19 |
|
|
| 20 |
# Repo root: must be a directory that contains groot/ (so experiment.py can be found). |
|
| 21 |
# Beaker/weka uses /root/yejink/dreamzero; image has /root/dreamzero; else script location. |
|
| 22 |
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)" |
|
| 23 |
SCRIPT_REPO_ROOT="$(cd "$SCRIPT_DIR/../.." && pwd)" |
|
| 24 |
if [ -n "$DREAMZERO_ROOT" ] && [ … |
|
| 31:17 |
…OT="$SCRIPT_REPO_ROOT" |
|
| 32 |
else |
|
| 33 |
DREAMZERO_ROOT="${DREAMZERO_ROOT:-/root/yejink/dreamzero}" |
|
| 34 |
fi |
|
| 35 |
if [ ! -d "$DREAMZERO_ROOT/groot" ]; then |
|
| 36 |
echo "ERROR: No groot/ under $DREAMZERO_ROOT. Set DREAMZERO_ROOT to the dreamzero repo root that contains groot/." |
|
| 37 |
exit 1 |
|
| 38 |
fi |
|
| 39 |
|
|
| 40 |
# ============ USER CONFIGURATION ============ |
|
| 41 |
NUM_GPUS=${NUM_GPUS:-8} |
|
| 42 |
DROID_DATA_ROOT=${DR… |
|
Identity
| SHA-256 |
ea275a797f89b011d48883fbe930db3b2ed244ab102bc9a7dd9df03aa1bdcb40 |
| Filename |
droid_training_wan22.sh |
| Package |
zhicao/dreamzero |
| Version |
main |
| PURL |
pkg:huggingface/zhicao/dreamzero@main |
Timeline
| First seen |
13 Aug 2026 21:18 UTC |
| First analyzed |
27 Aug 2026 17:12 UTC |
| Last analyzed |
27 Aug 2026 17:12 UTC |
| Last updated |
27 Aug 2026 17:12 UTC |
Labeling
| Label |
unknown |
| Label source |
forager |
| Traits version |
b59b7 |