aixsatoshi commited on
Commit
3f6e58a
1 Parent(s): 0e7ba44

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -42,7 +42,7 @@ model = AutoModelForCausalLM.from_pretrained(
42
  torch_dtype=torch.float16,
43
  device_map="auto",
44
  )
45
- tokenizer = AutoTokenizer.from_pretrained(MODELS)
46
 
47
  @spaces.GPU
48
  def stream_chat(message: str, history: list, temperature: float, max_new_tokens: int, top_p: float, top_k: int, penalty: float):
 
42
  torch_dtype=torch.float16,
43
  device_map="auto",
44
  )
45
+ tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
46
 
47
  @spaces.GPU
48
  def stream_chat(message: str, history: list, temperature: float, max_new_tokens: int, top_p: float, top_k: int, penalty: float):