Spaces:
Runtime error
Runtime error
jaymojnidar
commited on
Commit
•
598fcdf
1
Parent(s):
5bfa3aa
adding my data
Browse files
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']
|