Update README, add final results, add util scripts to query based on user input.
This commit is contained in:
@@ -8,7 +8,7 @@ from peft import PeftModel
|
||||
# Config
|
||||
# -----------------------------
|
||||
BASE_MODEL_NAME = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
|
||||
ADAPTER_PATH = "./ft_lora_adapter"
|
||||
ADAPTER_PATH = "./ft_lora2_adapter"
|
||||
|
||||
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
||||
|
||||
|
||||
Reference in New Issue
Block a user