Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -25,11 +25,11 @@ token = os.environ["HF_TOKEN"]
|
|
25 |
# beomi/Llama-3-KoEn-8B-Instruct-preview 모델과 토크나이저 같이 로드
|
26 |
model = AutoModelForCausalLM.from_pretrained(
|
27 |
# "microsoft/Phi-3-mini-128k-instruct",
|
28 |
-
"
|
29 |
token=token,
|
30 |
trust_remote_code=True,
|
31 |
)
|
32 |
-
tok = AutoTokenizer.from_pretrained("
|
33 |
|
34 |
# 종료 토큰 ID 설정
|
35 |
terminators = [
|
|
|
25 |
# beomi/Llama-3-KoEn-8B-Instruct-preview 모델과 토크나이저 같이 로드
|
26 |
model = AutoModelForCausalLM.from_pretrained(
|
27 |
# "microsoft/Phi-3-mini-128k-instruct",
|
28 |
+
"apple/OpenELM-3B-Instruct",
|
29 |
token=token,
|
30 |
trust_remote_code=True,
|
31 |
)
|
32 |
+
tok = AutoTokenizer.from_pretrained("apple/OpenELM-3B-Instruct", token=token)
|
33 |
|
34 |
# 종료 토큰 ID 설정
|
35 |
terminators = [
|