Fix model types map.

h2oai · May 11, 2023 · a491987 · a491987
1 parent ba6cad3
commit a491987
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/finetune.py b/finetune.py
@@ -56,8 +56,8 @@ class PromptType(Enum):
         'h2oai/h2ogpt-gm-oasst1-en-1024-20b',
         'h2oai/h2ogpt-gm-oasst1-en-1024-12b',
         'h2oai/h2ogpt-gm-oasst1-multilang-1024-20b',
-        'h2ogpt-gm-oasst1-en-2048-open-llama-7b-preview-300bt',
-        'h2ogpt-gm-oasst1-en-2048-open-llama-7b-preview-300bt-v2',
+        'h2oai/h2ogpt-gm-oasst1-en-2048-open-llama-7b-preview-300bt',
+        'h2oai/h2ogpt-gm-oasst1-en-2048-open-llama-7b-preview-300bt-v2',
     ],
     'instruct': [],
     'instruct_with_end': ['databricks/dolly-v2-12b'],