aswerdlow commited on
Commit
2d469e3
·
1 Parent(s): d92a774
Files changed (1) hide show
  1. demo/demo.sh +1 -1
demo/demo.sh CHANGED
@@ -6,6 +6,6 @@ ls -la /home/appuser/app/ckpts/unidisc_interleaved
6
  PYTHONUNBUFFERED=1 uv run python demo/client.py &
7
 
8
  PYTHONUNBUFFERED=1 UNIDISC_FORCE_CUDNN_SPDA_CONTEXT=1 uv run python demo/server.py experiments='[large_scale_train,large_scale_train_high_res_interleaved,eval_unified,large_scale_high_res_interleaved_inference]' \
9
- trainer.load_from_state_dict="/home/appuser/app/ckpts/unidisc_interleaved/unidisc_interleaved.pt" model.use_custom_vae_ckpt="/home/appuser/app/ckpts/unidisc_interleaved/vq_ds16_t2i.pt" &
10
 
11
  wait
 
6
  PYTHONUNBUFFERED=1 uv run python demo/client.py &
7
 
8
  PYTHONUNBUFFERED=1 UNIDISC_FORCE_CUDNN_SPDA_CONTEXT=1 uv run python demo/server.py experiments='[large_scale_train,large_scale_train_high_res_interleaved,eval_unified,large_scale_high_res_interleaved_inference]' \
9
+ trainer.load_from_state_dict="/home/appuser/app/ckpts/unidisc_interleaved/unidisc_interleaved.pt" &
10
 
11
  wait