dennyaw commited on
Commit
a757e69
1 Parent(s): ff68b4a

model first then tokenizer

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -3,8 +3,9 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
3
  import torch
4
 
5
  model_name="fadliaulawi/polylm-1.7b-finetuned"
6
- tokenizer = AutoTokenizer.from_pretrained(model_name)
7
  model = AutoModelForCausalLM.from_pretrained(model_name)
 
 
8
 
9
 
10
  def user(message, history):
 
3
  import torch
4
 
5
  model_name="fadliaulawi/polylm-1.7b-finetuned"
 
6
  model = AutoModelForCausalLM.from_pretrained(model_name)
7
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
8
+
9
 
10
 
11
  def user(message, history):