JUNGU commited on
Commit
98b100b
·
verified ·
1 Parent(s): 1868f23

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- "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 = [
 
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 = [