fix

edwko · Jan 15, 2025 · ad5ba96 · ad5ba96
1 parent 7f039a8
commit ad5ba96
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 11 deletions.
diff --git a/outetts/__init__.py b/outetts/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "0.3.1" 
+__version__ = "0.3.2" 
 
 from .interface import InterfaceHF, InterfaceGGUF, InterfaceEXL2, display_available_models
 from .interface import HFModelConfig_v1, GGUFModelConfig_v1, EXL2ModelConfig_v1

diff --git a/outetts/interface.py b/outetts/interface.py
@@ -100,9 +100,6 @@ def InterfaceHF(
 
     config = get_model_config(model_version)
 
-    if model_version not in ["0.1", "0.2", "0.3"]:
-        raise ValueError(f"Invalid model version!")
-
     if model_version == "0.3":
         if cfg.tokenizer_path is None:
             raise ValueError(f"tokenizer_path must be provided!")
@@ -145,9 +142,6 @@ def InterfaceGGUF(
 
     config = get_model_config(model_version)
 
-    if model_version not in ["0.1", "0.2", "0.3"]:
-        raise ValueError(f"Invalid model version!")
-
     if model_version == "0.3":
         if cfg.tokenizer_path is None:
             raise ValueError(f"tokenizer_path must be provided!")
@@ -186,9 +180,6 @@ def InterfaceEXL2(
 
     config = get_model_config(model_version)
 
-    if model_version not in ["0.1", "0.2", "0.3"]:
-        raise ValueError(f"Invalid model version!")
-
     if model_version == "0.3":
         if cfg.tokenizer_path is None:
             raise ValueError(f"tokenizer_path must be provided!")

diff --git a/setup.py b/setup.py
@@ -8,7 +8,7 @@
 
 setup(
     name='outetts',
-    version='0.3.1',
+    version='0.3.2',
     packages=find_packages(),
     install_requires=install_requires,
     author='OuteAI',