update

Dockerfile

@@ -64,7 +64,7 @@ COPY --chown=appuser third_party third_party
 COPY --chown=appuser ckpts ckpts
 COPY --chown=appuser ./__* ./
 COPY --chown=appuser ./*.py ./
-COPY --chown=appuser ./archive/pytorch_model_fsdp.bin ./
+# RUN huggingface-cli download aswerdlow/unidisc_interleaved --local-dir ./ckpts/unidisc_interleaved
 
 # Switch to non-root user
 USER appuser

demo/demo.sh

@@ -1,6 +1,6 @@
 #!/bin/bash
 
 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]' \
-trainer.load_from_state_dict="/home/appuser/app/pytorch_model_fsdp.bin" &
+trainer.load_from_state_dict="/home/appuser/app/ckpts/unidisc_interleaved/unidisc_interleaved.pt" &
 
 uv run python demo/client.py