diff --git a/scripts/6_train_SAEHD.sh b/scripts/6_train_SAEHD.sh
index 10ea707ef..fae493fca 100755
--- a/scripts/6_train_SAEHD.sh
+++ b/scripts/6_train_SAEHD.sh
@@ -6,5 +6,6 @@ $DFL_PYTHON "$DFL_SRC/main.py" train \
     --training-data-dst-dir "$DFL_WORKSPACE/data_dst/aligned" \
     --pretraining-data-dir "$DFL_ROOT/pretrain_CelebA" \
     --model-dir "$DFL_WORKSPACE/model" \
-    --model SAEHD
-
+    --model SAEHD \
+    --force-model-name DFD224F \
+    --force-gpu-idxs 0