Update app.py
Browse files
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
|