tevykuch commited on
Commit
089db6b
·
verified ·
1 Parent(s): d2afd0e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,8 +15,8 @@ generation_config = {
15
  "stop_token": '### Instruction:'
16
  }
17
 
18
- tokenizer = AutoTokenizer.from_pretrained("tevykuch/sl0th")
19
- llm = AutoModelForCausalLM.from_pretrained("tevykuch/sl0th")
20
 
21
  def stream(prompt):
22
  # Tokenize the prompt
 
15
  "stop_token": '### Instruction:'
16
  }
17
 
18
+ llm = AutoModelForCausalLM.from_pretrained("tevykuch/sl0th", hf=True)
19
+ tokenizer = AutoTokenizer.from_pretrained(llm)
20
 
21
  def stream(prompt):
22
  # Tokenize the prompt