Update s1_train.py

GPT_SoVITS/s1_train.py

@@ -79,6 +79,8 @@ class my_model_ckpt(ModelCheckpoint):
                     to_save_od["config"] = self.config
                     to_save_od["info"] = "GPT-e%s" % (trainer.current_epoch + 1)
                     # torch.save(
+                    # print(os.environ)
+                    if(os.environ.get("LOCAL_RANK","0")=="0"):
                         my_save(
                             to_save_od,
                             "%s/%s-e%s.ckpt"