mamkkl commited on
Commit
f405f85
·
verified ·
1 Parent(s): 608fa3f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -82,7 +82,9 @@ def loadModel():
82
  lora_weights,
83
  device_map="auto",
84
  cache_dir=cache_dir,
85
- torch_dtype=torch.float16
 
 
86
  )
87
  model.eval()
88
  return model
 
82
  lora_weights,
83
  device_map="auto",
84
  cache_dir=cache_dir,
85
+ torch_dtype=torch.float16,
86
+ assign=True,
87
+ is_trainable=False,
88
  )
89
  model.eval()
90
  return model