ronald commited on
Commit
6c6129a
1 Parent(s): aef05c0
Files changed (1) hide show
  1. ccl_win.py +1 -0
ccl_win.py CHANGED
@@ -20,6 +20,7 @@ from transformers import AutoModelForSequenceClassification, AutoTokenizer
20
  import getpass
21
  import pdb
22
  import os
 
23
 
24
  # TODO: Add BibTeX citation
25
  _CITATION = """\
 
20
  import getpass
21
  import pdb
22
  import os
23
+ import torch
24
 
25
  # TODO: Add BibTeX citation
26
  _CITATION = """\