Update pandrator_installer_launcher.py

lukaszliniewicz · Nov 11, 2024 · 9745799 · 9745799
1 parent 104f3eb
commit 9745799
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pandrator_installer_launcher.py b/pandrator_installer_launcher.py
@@ -195,7 +195,7 @@ def install_pytorch_for_xtts_finetuning(self, conda_path, env_name):
             self.run_command([
                 os.path.join(conda_path, 'Scripts', 'conda.exe'),
                 'run', '-p', env_path,
-                'pip', 'install', 'torch==2.1.1+cu118', 'torchaudio==2.1.1+cu118',
+                'pip', 'install', 'torch==2.2.0+cu118', 'torchaudio==2.2.0+cu118',
                 '--index-url', 'https://download.pytorch.org/whl/cu118'
             ])
             logging.info("PyTorch for XTTS Fine-tuning installed successfully.")