Spaces:
Runtime error
Runtime error
O S I H
commited on
Commit
•
abfec2c
1
Parent(s):
70f98c5
edit
Browse files
app.py
CHANGED
|
@@ -22,10 +22,11 @@ token = os.environ["HF_TOKEN"]
|
|
| 22 |
|
| 23 |
model = AutoModelForCausalLM.from_pretrained(
|
| 24 |
"microsoft/Phi-3-small-8k-instruct",
|
| 25 |
-
token=token,
|
|
|
|
| 26 |
trust_remote_code=True,
|
| 27 |
)
|
| 28 |
-
tok = AutoTokenizer.from_pretrained("microsoft/Phi-3-small-8k-instruct"
|
| 29 |
terminators = [
|
| 30 |
tok.eos_token_id,
|
| 31 |
]
|
|
|
|
| 22 |
|
| 23 |
model = AutoModelForCausalLM.from_pretrained(
|
| 24 |
"microsoft/Phi-3-small-8k-instruct",
|
| 25 |
+
# token=token,
|
| 26 |
+
torch_dtype="auto",
|
| 27 |
trust_remote_code=True,
|
| 28 |
)
|
| 29 |
+
tok = AutoTokenizer.from_pretrained("microsoft/Phi-3-small-8k-instruct")#, token=token
|
| 30 |
terminators = [
|
| 31 |
tok.eos_token_id,
|
| 32 |
]
|