Subi003 commited on
Commit
d4065b4
·
verified ·
1 Parent(s): 0a9ac4d

Update modeling.py

Browse files
Files changed (1) hide show
  1. modeling.py +5 -5
modeling.py CHANGED
@@ -50,8 +50,8 @@ class EmoAxis(PreTrainedModel):
50
  return logits
51
 
52
 
53
- # ============================================================
54
- # REGISTER WITH HUGGINGFACE AUTOCLASSES
55
- # ============================================================
56
- AutoConfig.register("emoaxis", EmoAxisConfig)
57
- AutoModel.register(EmoAxisConfig, EmoAxis)
 
50
  return logits
51
 
52
 
53
+ # --- Registration for AutoModel ---
54
+ try:
55
+ AutoModel.register(AutoConfig, EmoAxis)
56
+ except:
57
+ pass