vickeee465 commited on
Commit
7edf01a
·
1 Parent(s): 41bc8d2
Files changed (1) hide show
  1. utils.py +2 -2
utils.py CHANGED
@@ -23,8 +23,8 @@ def download_hf_models():
23
  for model_id in models:
24
  model = AutoModelForSequenceClassification.from_pretrained(model_id, low_cpu_mem_usage=True, device_map="auto",
25
  token=HF_TOKEN)
26
- del model
27
  for tokenizer_id in tokenizers:
28
  tokenizer = AutoTokenizer.from_pretrained(tokenizer_id)
29
- del tokenizer
30
 
 
23
  for model_id in models:
24
  model = AutoModelForSequenceClassification.from_pretrained(model_id, low_cpu_mem_usage=True, device_map="auto",
25
  token=HF_TOKEN)
26
+ del model
27
  for tokenizer_id in tokenizers:
28
  tokenizer = AutoTokenizer.from_pretrained(tokenizer_id)
29
+ del tokenizer
30