adding my data

model.py

@@ -10,6 +10,7 @@ from huggingface_hub import login
 #model_id = 'meta-llama/Llama-2-7b-chat-hf'
 #model_id = 'Trelis/Llama-2-7b-chat-hf-sharded-bf16'
 model_id = 'jaymojnidar/llama2-finetuned-mydata'
+confg_model_id = 'Trelis/Llama-2-7b-chat-hf-sharded-bf16'
 
 if torch.cuda.is_available():
     tok = os.environ['HF_TOKEN']
@@ -20,7 +21,7 @@ if torch.cuda.is_available():
           #, token="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"
           )
     
-    config = AutoConfig.from_pretrained(model_id, use_auth_token=True)
+    config = AutoConfig.from_pretrained(confg_model_id, use_auth_token=True)
     config.pretraining_tp = 1
     model = AutoModelForCausalLM.from_pretrained(
         model_id,