Update app.py
Browse files
app.py
CHANGED
@@ -83,7 +83,6 @@ def loadModel():
|
|
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()
|
|
|
83 |
device_map="auto",
|
84 |
cache_dir=cache_dir,
|
85 |
torch_dtype=torch.float16,
|
|
|
86 |
is_trainable=False,
|
87 |
)
|
88 |
model.eval()
|