diff --git a/Dockerfile b/Dockerfile
index aaf5067..a92d9e9 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -87,7 +87,7 @@ EOF
 source ~/.bashrc
 export TRAINER_TELEMETRY=0
 conda activate alltalk
-python -m trainer.distribute --script finetune.py
+python finetune.py
 EOF
     cat << EOF > start_diagnostics.sh
 #!/usr/bin/env bash
diff --git a/finetune.py b/finetune.py
index e84ebf0..7fdba69 100644
--- a/finetune.py
+++ b/finetune.py
@@ -4805,5 +4805,5 @@ def train_model(
         share=False,
         debug=False,
         server_port=7052,
-        server_name="127.0.0.1",
+        server_name="0.0.0.0",
     )