Locutusque commited on
Commit
70111fb
·
verified ·
1 Parent(s): 0d5ca2a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -22,8 +22,6 @@ def generate(
22
  ):
23
  try:
24
  pipe = load_model(model_name)
25
- if model_name == "Locutusque/Hercules-6.1-Llama-3.1-8B":
26
- model_name = "meta-llama/Llama-3.1-8B-Instruct"
27
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True, token=os.environ["token"])
28
  tokenizer.eos_token = "<|im_end|>"
29
  print(tokenizer)
 
22
  ):
23
  try:
24
  pipe = load_model(model_name)
 
 
25
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True, token=os.environ["token"])
26
  tokenizer.eos_token = "<|im_end|>"
27
  print(tokenizer)