From d0ebce6bacbc78d7ac642169d0d6afec4c2209f2 Mon Sep 17 00:00:00 2001
From: mrq <mrq@ecker.tech>
Date: Sat, 1 Jun 2024 10:30:13 -0500
Subject: [PATCH] ugh

---
 vall_e/data.py          | 4 ++--
 vall_e/utils/trainer.py | 2 ++
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/vall_e/data.py b/vall_e/data.py
index 3bea75b..8122e65 100755
--- a/vall_e/data.py
+++ b/vall_e/data.py
@@ -242,8 +242,8 @@ class Dataset(_Dataset):
 			# recreate paths_by_spkr_name
 			self.paths_by_spkr_name = {}
 			for path in self.paths:
-				name = cfg.get_spkr( path )
-				if name not in self.paths_by_spkr_name[name]:
+				name = cfg.get_spkr( Path(path) )
+				if name not in self.paths_by_spkr_name:
 					self.paths_by_spkr_name[name] = []
 				self.paths_by_spkr_name[name].append( path )
 
diff --git a/vall_e/utils/trainer.py b/vall_e/utils/trainer.py
index b25eeb6..5197ce0 100755
--- a/vall_e/utils/trainer.py
+++ b/vall_e/utils/trainer.py
@@ -150,8 +150,10 @@ def train(
 	last_save_step = engines.global_step
 	last_eval_step = 0
 
+	"""
 	if cfg.distributed:
 		train_dl.sampler.set_epoch(int(engines.global_samples / len(train_dl.dataset.paths)))
+	"""
 
 	# Training loop
 	for batch in _make_infinite_epochs(train_dl):