Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,14 @@ from accelerate import Accelerator
|
|
11 |
hf_api_key = os.getenv('HF_API_KEY')
|
12 |
|
13 |
# λͺ¨λΈ ID λ° ν ν¬λμ΄μ μ€μ
|
|
|
14 |
model_id = "microsoft/Phi-3-mini-128k-instruct"
|
|
|
|
|
|
|
|
|
|
|
|
|
15 |
tokenizer = AutoTokenizer.from_pretrained(model_id, token=hf_api_key)
|
16 |
accelerator = Accelerator()
|
17 |
|
|
|
11 |
hf_api_key = os.getenv('HF_API_KEY')
|
12 |
|
13 |
# λͺ¨λΈ ID λ° ν ν¬λμ΄μ μ€μ
|
14 |
+
# λͺ¨λΈ ID
|
15 |
model_id = "microsoft/Phi-3-mini-128k-instruct"
|
16 |
+
|
17 |
+
# μ¬μ©μ μ μ μ½λλ₯Ό μ λ’°νκ³ μ€ννλλ‘ μ€μ
|
18 |
+
model = AutoModelForCausalLM.from_pretrained(
|
19 |
+
model_id,
|
20 |
+
trust_remote_code=True # μ¬μ©μ μ μ μ½λ μ€ν νμ©
|
21 |
+
)
|
22 |
tokenizer = AutoTokenizer.from_pretrained(model_id, token=hf_api_key)
|
23 |
accelerator = Accelerator()
|
24 |
|