Spaces:
Running
on
Zero
Running
on
Zero
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():
|