Update app.py
Browse files
app.py
CHANGED
|
@@ -65,8 +65,7 @@ def main(
|
|
| 65 |
)
|
| 66 |
model = PeftModel.from_pretrained(
|
| 67 |
model,
|
| 68 |
-
|
| 69 |
-
adapter_name=lora_weights,
|
| 70 |
device_map={"": device},
|
| 71 |
)
|
| 72 |
|
|
|
|
| 65 |
)
|
| 66 |
model = PeftModel.from_pretrained(
|
| 67 |
model,
|
| 68 |
+
lora_weights,
|
|
|
|
| 69 |
device_map={"": device},
|
| 70 |
)
|
| 71 |
|