natalieparker commited on
Commit
d393ff4
Β·
verified Β·
1 Parent(s): 7f4a37d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ app = FastAPI()
7
  MODEL = "natalieparker/LumaAI-160M-v3"
8
 
9
  print("πŸ”„ Loading tokenizer...")
10
- tokenizer = AutoTokenizer.from_pretrained(MODEL)
11
 
12
  print("πŸ”„ Loading model on CPU...")
13
  model = AutoModelForCausalLM.from_pretrained(
 
7
  MODEL = "natalieparker/LumaAI-160M-v3"
8
 
9
  print("πŸ”„ Loading tokenizer...")
10
+ tokenizer = AutoTokenizer.from_pretrained(MODEL, use_fast=True)
11
 
12
  print("πŸ”„ Loading model on CPU...")
13
  model = AutoModelForCausalLM.from_pretrained(