nehalelkaref commited on
Commit
fd5008e
·
1 Parent(s): 1a9bdae

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +4 -4
main.py CHANGED
@@ -14,11 +14,11 @@ fusion_adapter = Repository(local_dir="fusion_adapter", clone_from="nehalelkaref
14
 
15
  model = AutoAdapterModel.from_pretrained("UBC-NLP/MARBERT")
16
 
17
- model.load_adapter("/code/aoc3_adapter", set_active=True, with_head=False)
18
- model.load_adapter("/code/aoc4_adapter", set_active=True, with_head=False)
19
- model.load_adapter("/code/sarcasm_adapter", set_active=True, with_head=False)
20
 
21
- model.load_adapter_fusion("/code/fusion_adapter/aoc(3),aoc(4),sarcasm",with_head=True, set_active=True)
22
 
23
  pipe = TextClassificationPipeline(tokenizer=tokenizer, model=model)
24
 
 
14
 
15
  model = AutoAdapterModel.from_pretrained("UBC-NLP/MARBERT")
16
 
17
+ model.load_adapter("/aoc3_adapter", set_active=True, with_head=False)
18
+ model.load_adapter("/aoc4_adapter", set_active=True, with_head=False)
19
+ model.load_adapter("/sarcasm_adapter", set_active=True, with_head=False)
20
 
21
+ model.load_adapter_fusion("/fusion_adapter/aoc(3),aoc(4),sarcasm",with_head=True, set_active=True)
22
 
23
  pipe = TextClassificationPipeline(tokenizer=tokenizer, model=model)
24