alexkueck commited on
Commit
cc91f22
·
1 Parent(s): 4c0d9cc

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -0
utils.py CHANGED
@@ -23,6 +23,7 @@ import datasets
23
  from datasets import load_dataset
24
  import evaluate
25
  from transformers import LlamaForCausalLM, LlamaTokenizer
 
26
 
27
 
28
 
 
23
  from datasets import load_dataset
24
  import evaluate
25
  from transformers import LlamaForCausalLM, LlamaTokenizer
26
+ from setfit import SetFitModel, SetFitTrainer
27
 
28
 
29