newtechdevng commited on
Commit
3ef172b
·
verified ·
1 Parent(s): 3f4fbe9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -5,11 +5,16 @@ import torch
5
  MODEL_ID = "newtechdevng/math-tutor-smollm2-360M"
6
 
7
  print("Loading model...")
8
- tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, use_fast=False) # ← fix
 
 
 
 
9
  model = AutoModelForCausalLM.from_pretrained(
10
  MODEL_ID,
11
  torch_dtype=torch.float32,
12
- device_map="cpu"
 
13
  )
14
  model.eval()
15
  print("Ready!")
 
5
  MODEL_ID = "newtechdevng/math-tutor-smollm2-360M"
6
 
7
  print("Loading model...")
8
+ tokenizer = AutoTokenizer.from_pretrained(
9
+ MODEL_ID,
10
+ use_fast=False,
11
+ trust_remote_code=True
12
+ )
13
  model = AutoModelForCausalLM.from_pretrained(
14
  MODEL_ID,
15
  torch_dtype=torch.float32,
16
+ device_map="cpu",
17
+ trust_remote_code=True
18
  )
19
  model.eval()
20
  print("Ready!")