hosseinhimself commited on
Commit
5d8e9c5
·
verified ·
1 Parent(s): df2a3a5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ device = torch.device("cpu") # Ensure it's using CPU only
7
 
8
  # Load model and tokenizer
9
  model_name = "hosseinhimself/ISANG-v1.0-8B" # Replace with your model name
10
- tokenizer = AutoTokenizer.from_pretrained(model_name)
11
 
12
  # Load the model for inference on CPU
13
  model = AutoModelForCausalLM.from_pretrained(model_name).to(device)
 
7
 
8
  # Load model and tokenizer
9
  model_name = "hosseinhimself/ISANG-v1.0-8B" # Replace with your model name
10
+ tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=False)
11
 
12
  # Load the model for inference on CPU
13
  model = AutoModelForCausalLM.from_pretrained(model_name).to(device)