Fix: check hasattr for sampler.set_epoch on WeightedRandomSampler

train.py

@@ -701,7 +701,7 @@ def train(
     accum = max(1, grad_accum)
 
     for epoch in range(start_epoch, epochs + 1):
-        if sampler is not None:
+        if is_ddp and sampler is not None and hasattr(sampler, "set_epoch"):
             sampler.set_epoch(epoch)
         epoch_loss = 0.0
         for batch_idx, batch in enumerate(dataloader):

upload_to_hf.py

@@ -64,7 +64,7 @@ def upload_repository(repo_name: str = "ViuAI_TTS_200M", private: bool = False):
             ".git/*",
             "*.log",
         ],
-        commit_message="Performance: 32 parallel download workers and hf_transfer for 10x faster dataset downloading",
+        commit_message="Fix: check hasattr for sampler.set_epoch on WeightedRandomSampler",
     )
 
     print("\n" + "=" * 65)