File size: 459 Bytes
246c106
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
#!/usr/bin/bash

python3 -m venv venv
source venv/bin/activate
python -m pip install -r requirements.txt
FLASH_ATTENTION_SKIP_CUDA_BUILD=TRUE python -m pip install flash-attn==2.5.8 --no-build-isolation

# Download datasets to data/train_v1.0, data/val_v1.0
huggingface-cli download 1x-technologies/worldmodel --repo-type dataset --local-dir data

mv data/val_v1.1 data/1x_humanoid_magvit_traj1000_val
mv data/train_v1.1 data/1x_humanoid_magvit_traj1000_train