Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -35,7 +35,7 @@ def load_llm():
|
|
35 |
|
36 |
model_name = 'mistralai/Mistral-7B-Instruct-v0.2'
|
37 |
|
38 |
-
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
39 |
model = AutoModelForCausalLM.from_pretrained(
|
40 |
model_name,
|
41 |
#load_in_4bit=True,
|
@@ -43,6 +43,7 @@ def load_llm():
|
|
43 |
torch_dtype=torch.bfloat16,
|
44 |
device_map="auto",
|
45 |
trust_remote_code=True,
|
|
|
46 |
)
|
47 |
|
48 |
pipe = pipeline(
|
|
|
35 |
|
36 |
model_name = 'mistralai/Mistral-7B-Instruct-v0.2'
|
37 |
|
38 |
+
tokenizer = AutoTokenizer.from_pretrained(model_name, token=HF_TOKEN)
|
39 |
model = AutoModelForCausalLM.from_pretrained(
|
40 |
model_name,
|
41 |
#load_in_4bit=True,
|
|
|
43 |
torch_dtype=torch.bfloat16,
|
44 |
device_map="auto",
|
45 |
trust_remote_code=True,
|
46 |
+
token=HF_TOKEN,
|
47 |
)
|
48 |
|
49 |
pipe = pipeline(
|