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