darpanaswal commited on
Commit
65afd51
·
verified ·
1 Parent(s): fa0ca66

Update download_model.py

Browse files
Files changed (1) hide show
  1. download_model.py +2 -1
download_model.py CHANGED
@@ -1,11 +1,12 @@
1
  import torch
2
- import huggingface_hub import login
3
  from transformers import (AutoTokenizer,
4
  BitsAndBytesConfig,
5
  MBart50TokenizerFast,
6
  AutoModelForSeq2SeqLM,
7
  MBartForConditionalGeneration)
8
  from config import hf_token
 
9
 
10
  device = "cuda" if torch.cuda.is_available() else "cpu"
11
 
 
1
  import torch
2
+ import huggingface_hub
3
  from transformers import (AutoTokenizer,
4
  BitsAndBytesConfig,
5
  MBart50TokenizerFast,
6
  AutoModelForSeq2SeqLM,
7
  MBartForConditionalGeneration)
8
  from config import hf_token
9
+ from huggingface_hub import login
10
 
11
  device = "cuda" if torch.cuda.is_available() else "cpu"
12