JUNGU commited on
Commit
1868f23
·
verified ·
1 Parent(s): 33cc946

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
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("microsoft/Phi-3-mini-128k-instruct", token=token)
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 = [