Spaces:
Runtime error
Runtime error
harsh-manvar
commited on
Commit
•
3da0af1
1
Parent(s):
d40f94b
Update model.py
Browse files
model.py
CHANGED
@@ -16,8 +16,7 @@ device = "cpu"
|
|
16 |
|
17 |
|
18 |
model = AutoModelForCausalLM.from_pretrained(model_id, model_type="llama", lib='avx2', hf=True)
|
19 |
-
tokenizer = AutoTokenizer.from_pretrained('Llama-2-7b-chat-hf')
|
20 |
-
#tokenizer = AutoTokenizer.from_pretrained('meta-llama/Llama-2-7b-chat-hf')
|
21 |
|
22 |
def get_prompt(message: str, chat_history: list[tuple[str, str]],
|
23 |
system_prompt: str) -> str:
|
|
|
16 |
|
17 |
|
18 |
model = AutoModelForCausalLM.from_pretrained(model_id, model_type="llama", lib='avx2', hf=True)
|
19 |
+
tokenizer = AutoTokenizer.from_pretrained('meta-llama/Llama-2-7b-chat-hf')
|
|
|
20 |
|
21 |
def get_prompt(message: str, chat_history: list[tuple[str, str]],
|
22 |
system_prompt: str) -> str:
|