update version and dependencies

parler_tts/__init__.py

+__version__ = "0.1"
+
+
 from .configuration_parler_tts import ParlerTTSConfig, ParlerTTSDecoderConfig
 from .modeling_parler_tts import (
     ParlerTTSForCausalLM,

setup.py

-# Copyright 2023 The HuggingFace Team. All rights reserved.
+# Copyright 2024 The HuggingFace Team. All rights reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -12,21 +12,15 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-
 import os
-
 import setuptools
 
 
 _deps = [
     "transformers>=4.34.0",
-    "datasets[audio]>=2.14.5",
     "torch",
-    "accelerate",
-    "evaluate",
     "sentencepiece",
     "descript-audio-codec",
-    "jiwer",
 ]
 
 _extras_dev_deps = [
@@ -35,6 +29,14 @@ _extras_dev_deps = [
     "ruff>=0.0.241,<=0.0.259",
 ]
 
+_extras_training_deps = [
+    "jiwer",
+    "wandb",
+    "accelerate",
+    "evaluate",
+    "datasets[audio]>=2.14.5",
+]
+
 here = os.path.abspath(os.path.dirname(__file__))
 
 with open(os.path.join(here, "README.md"), encoding="utf-8") as f:
@@ -59,5 +61,6 @@ setuptools.setup(
     install_requires=_deps,
     extras_require={
         "dev": [_extras_dev_deps],
+        "train": [_extras_training_deps],
     },
 )