Hugging Face's logo

Spaces:
AbstractPhil
/
bert-beatrix-2048-testing
Running on Zero

App Files Community
AbstractPhil commited on
Commit
bbf3f7c
·
verified ·
1 Parent(s): 785df91

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -54,6 +54,10 @@ REPO_ID = "AbstractPhil/bert-beatrix-2048"
54
  REVISION = "main"
55
  tokenizer = AutoTokenizer.from_pretrained(REPO_ID, revision=REVISION)
56
  model = AutoModelForMaskedLM.from_pretrained(
 
 
 
 
57
  REPO_ID,
58
  revision=REVISION,
59
  trust_remote_code=True
 
54
  REVISION = "main"
55
  tokenizer = AutoTokenizer.from_pretrained(REPO_ID, revision=REVISION)
56
  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