Update app.py
Browse files
app.py
CHANGED
|
@@ -57,10 +57,7 @@ model = AutoModelForMaskedLM.from_pretrained(
|
|
| 57 |
REPO_ID,
|
| 58 |
revision=REVISION,
|
| 59 |
trust_remote_code=True,
|
| 60 |
-
ignore_mismatched_sizes=True
|
| 61 |
-
REPO_ID,
|
| 62 |
-
revision=REVISION,
|
| 63 |
-
trust_remote_code=True
|
| 64 |
).eval().cuda()
|
| 65 |
|
| 66 |
def build_interface():
|
|
|
|
| 57 |
REPO_ID,
|
| 58 |
revision=REVISION,
|
| 59 |
trust_remote_code=True,
|
| 60 |
+
ignore_mismatched_sizes=True
|
|
|
|
|
|
|
|
|
|
| 61 |
).eval().cuda()
|
| 62 |
|
| 63 |
def build_interface():
|