Update my_model/LLAMA2/LLAMA2_model.py

my_model/LLAMA2/LLAMA2_model.py

@@ -3,7 +3,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
 from typing import Optional
 import bitsandbytes  # only for using on GPU
 import accelerate  # only for using on GPU
-from my_model.LLAMA2 import LLAMA2_config as config  # Importing LLAMA2 configuration file
+from my_model.config import LLAMA2_config as config  # Importing LLAMA2 configuration file
 import warnings
 
 # Suppress only FutureWarning from transformers