Update better default mesh_dim

Treemann · Feb 21, 2024 · 5772bba · 5772bba
1 parent 0f441d3
commit 5772bba
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/run_sample_image.sh b/scripts/run_sample_image.sh
@@ -18,7 +18,7 @@ python3 -u -m lwm.vision_generation \
  --cfg_scale_image=5.0 \
  --vqgan_checkpoint="$vqgan_checkpoint" \
  --n_frames=1 \
- --mesh_dim='!-1,1,8,1' \
+ --mesh_dim='!1,1,-1,1' \
  --dtype='fp32' \
  --load_llama_config='7b' \
  --update_llama_config="dict(sample_mode='vision',theta=50000000,max_sequence_length=32768,use_flash_attention=True,scan_attention=False,scan_query_chunk_size=128,scan_key_chunk_size=128,scan_mlp=False,scan_mlp_chunk_size=8192,scan_layers=True)" \