TenzinGayche commited on
Commit
7a8369a
·
verified ·
1 Parent(s): 1a28685

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ model = AutoModelForCausalLM.from_pretrained(path, torch_dtype=torch.float16).to
21
 
22
  model.config.sliding_window = 4096
23
  model.eval()
24
-
25
  # Shared stop event
26
  stop_event = Event()
27
 
 
21
 
22
  model.config.sliding_window = 4096
23
  model.eval()
24
+ model.config.use_cache = True
25
  # Shared stop event
26
  stop_event = Event()
27