zkmine commited on
Commit
ce706e2
·
verified ·
1 Parent(s): a624f7a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -57,7 +57,8 @@ print("Loading model (this can take a minute on first launch)...")
57
  DEVICE = _device()
58
  DTYPE = torch.float16 if DEVICE in ("cuda", "mps") else torch.float32
59
 
60
- tokenizer = AutoTokenizer.from_pretrained(ADAPTER_DIR, trust_remote_code=True)
 
61
  tokenizer.pad_token = tokenizer.eos_token
62
 
63
  _base = AutoModelForCausalLM.from_pretrained(
 
57
  DEVICE = _device()
58
  DTYPE = torch.float16 if DEVICE in ("cuda", "mps") else torch.float32
59
 
60
+ tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, trust_remote_code=True)
61
+
62
  tokenizer.pad_token = tokenizer.eos_token
63
 
64
  _base = AutoModelForCausalLM.from_pretrained(