Update my_model/config/LLAMA2_config.py

my_model/config/LLAMA2_config.py

@@ -2,7 +2,7 @@
 import torch
 import os
 
-MODEL_NAME = "meta-llama/Llama-2-7b-chat-hf"
+MODEL_NAME = "meta-llama/Llama-2-7b-chat-hf"  # For loading and fine-tuning the 13b version --> MODEL_NAME = "meta-llama/Llama-2-13b-chat-hf"
 TOKENIZER_NAME = "meta-llama/Llama-2-7b-chat-hf"
 QUANTIZATION = '4bit'  # Options: '4bit', '8bit', or None
 FROM_SAVED = False