kovacsvi commited on
Commit
3abd99d
·
1 Parent(s): f17fb84

parameters removed

Browse files
Files changed (1) hide show
  1. utils.py +4 -4
utils.py CHANGED
@@ -54,7 +54,7 @@ for domain in domains_illframes.values():
54
 
55
  tokenizers = ["xlm-roberta-large"]
56
 
57
- def download_hf_models(models=[], tokenizers=[], hf_token=None):
58
  # Ensure the JIT model directory exists
59
  os.makedirs(JIT_DIR, exist_ok=True)
60
 
@@ -64,12 +64,12 @@ def download_hf_models(models=[], tokenizers=[], hf_token=None):
64
  # Load model and tokenizer
65
  model = AutoModelForSequenceClassification.from_pretrained(
66
  model_id,
67
- token=hf_token,
68
  device_map="auto"
69
  )
70
  tokenizer = AutoTokenizer.from_pretrained(
71
  model_id,
72
- token=hf_token
73
  )
74
 
75
  model.eval()
@@ -99,7 +99,7 @@ def download_hf_models(models=[], tokenizers=[], hf_token=None):
99
  print(f"Downloading tokenizer: {tokenizer_id}")
100
  AutoTokenizer.from_pretrained(
101
  tokenizer_id,
102
- token=hf_token
103
  )
104
 
105
  def df_h():
 
54
 
55
  tokenizers = ["xlm-roberta-large"]
56
 
57
+ def download_hf_models():
58
  # Ensure the JIT model directory exists
59
  os.makedirs(JIT_DIR, exist_ok=True)
60
 
 
64
  # Load model and tokenizer
65
  model = AutoModelForSequenceClassification.from_pretrained(
66
  model_id,
67
+ token=HF_TOKEN,
68
  device_map="auto"
69
  )
70
  tokenizer = AutoTokenizer.from_pretrained(
71
  model_id,
72
+ token=HF_TOKEN
73
  )
74
 
75
  model.eval()
 
99
  print(f"Downloading tokenizer: {tokenizer_id}")
100
  AutoTokenizer.from_pretrained(
101
  tokenizer_id,
102
+ token=HF_TOKEN
103
  )
104
 
105
  def df_h():