Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ def load_model():
|
|
9 |
model_id = "microsoft/phi-2"
|
10 |
access_token = os.environ.get("HF_AUTH_TOKEN")
|
11 |
|
12 |
-
tokenizer = AutoTokenizer.from_pretrained(model_id,
|
13 |
model = AutoModelForCausalLM.from_pretrained(
|
14 |
model_id,
|
15 |
device_map="auto",
|
|
|
9 |
model_id = "microsoft/phi-2"
|
10 |
access_token = os.environ.get("HF_AUTH_TOKEN")
|
11 |
|
12 |
+
tokenizer = AutoTokenizer.from_pretrained(model_id, token=access_token)
|
13 |
model = AutoModelForCausalLM.from_pretrained(
|
14 |
model_id,
|
15 |
device_map="auto",
|