ulduldp commited on
Commit
54700cc
·
verified ·
1 Parent(s): c70d6d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -13,13 +13,14 @@ HF_TOKEN = os.getenv("HF_TOKEN")
13
  login(token=HF_TOKEN)
14
 
15
  print("Loading tokenizer...")
16
- tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
17
 
18
  print("Loading model...")
19
  model = AutoModelForCausalLM.from_pretrained(
20
  MODEL_ID,
21
  torch_dtype=torch.float32,
22
- device_map="cpu"
 
23
  )
24
 
25
  print("Model loaded!")
 
13
  login(token=HF_TOKEN)
14
 
15
  print("Loading tokenizer...")
16
+ tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, token=HF_TOKEN)
17
 
18
  print("Loading model...")
19
  model = AutoModelForCausalLM.from_pretrained(
20
  MODEL_ID,
21
  torch_dtype=torch.float32,
22
+ device_map="cpu",
23
+ token=HF_TOKEN
24
  )
25
 
26
  print("Model loaded!")