Safetensors
custom_code
jonggwon-park commited on
Commit
3f2d37f
·
1 Parent(s): 78003e8

add model.eval()

Browse files
Files changed (1) hide show
  1. inference.py +1 -0
inference.py CHANGED
@@ -20,6 +20,7 @@ def load_model(device, dtype):
20
  torch_dtype=dtype,
21
  device_map=device,
22
  )
 
23
 
24
  models = {
25
  "tokenizer": tokenizer,
 
20
  torch_dtype=dtype,
21
  device_map=device,
22
  )
23
+ model.eval()
24
 
25
  models = {
26
  "tokenizer": tokenizer,