Create h2oai/h2ogpt-oasst1-512-20b

h2oai · Apr 23, 2023 · e0202c5 · e0202c5
1 parent 7154c64
commit e0202c5
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/export_hf_checkpoint.py b/export_hf_checkpoint.py
@@ -7,9 +7,9 @@
 from transformers import PreTrainedModel
 from finetune import get_loaders
 
-BASE_MODEL = 'h2oai/h2ogpt-oig-oasst1-512-6.9b'
-LORA_WEIGHTS = 'h2ogpt-oig-oasst1-512-6.9b.h2oaiopenassistant_oasst1_h2ogpt.2_epochs.e35e2e06e0af2f7dceac2e16e3646c90ccce4ec0.1'
-OUTPUT_NAME = "h2ogpt-oig-oasst1-512-6.9b"
+BASE_MODEL = 'h2oai/h2ogpt-oasst1-512-20b'
+LORA_WEIGHTS = 'h2ogpt-oasst1-512-20b.h2oaiopenassistant_oasst1_h2ogpt.2_epochs.fcaae7ef70600de8c97c9b38cb3f0075467cdad1.3'
+OUTPUT_NAME = "h2ogpt-oig-oasst1-512-20b"
 llama_type = "llama" in BASE_MODEL
 as_pytorch = False  # False -> HF