jaymojnidar commited on
Commit
598fcdf
1 Parent(s): 5bfa3aa

adding my data

Browse files
Files changed (1) hide show
  1. model.py +2 -1
model.py CHANGED
@@ -8,7 +8,8 @@ from huggingface_hub import login
8
 
9
  # model_id = 'meta-llama/Llama-2-13b-chat-hf'
10
  #model_id = 'meta-llama/Llama-2-7b-chat-hf'
11
- model_id = 'Trelis/Llama-2-7b-chat-hf-sharded-bf16'
 
12
 
13
  if torch.cuda.is_available():
14
  tok = os.environ['HF_TOKEN']
 
8
 
9
  # model_id = 'meta-llama/Llama-2-13b-chat-hf'
10
  #model_id = 'meta-llama/Llama-2-7b-chat-hf'
11
+ #model_id = 'Trelis/Llama-2-7b-chat-hf-sharded-bf16'
12
+ model_id = 'jaymojnidar/llama2-finetuned-mydata'
13
 
14
  if torch.cuda.is_available():
15
  tok = os.environ['HF_TOKEN']