rajyalakshmijampani commited on
Commit
698ef0c
·
verified ·
1 Parent(s): dea18de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -30,6 +30,10 @@ model = AutoModel.from_pretrained(
30
  token=HF_TOKEN
31
  ).to(device)
32
 
 
 
 
 
33
  model.eval()
34
 
35
  tokenizer = AutoTokenizer.from_pretrained(
 
30
  token=HF_TOKEN
31
  ).to(device)
32
 
33
+ # Patch missing attribute
34
+ if not hasattr(model.config, "pad_token_id"):
35
+ model.config.pad_token_id = 0
36
+
37
  model.eval()
38
 
39
  tokenizer = AutoTokenizer.from_pretrained(