update lora model id in correctness test (deepjavalibrary#1263)

maaquib · Nov 3, 2023 · a6fa50e · a6fa50e
1 parent 7660014
commit a6fa50e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/integration/llm/unmerged_lora.py b/tests/integration/llm/unmerged_lora.py
@@ -5,7 +5,7 @@
 import json
 import logging
 
-BASE_MODEL_ID = "decapoda-research/llama-7b-hf"
+BASE_MODEL_ID = "huggyllama/llama-7b"
 LORA_ADAPTER_1_ID = "tloen/alpaca-lora-7b"
 LORA_ADAPTER_2_ID = "22h/cabrita-lora-v0-1"
 LORA_ADAPTER_1_NAME = "english-alpaca"