tanyuzhou commited on
Commit
833d7c1
1 Parent(s): f5ef79f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -4,8 +4,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
4
 
5
  import spaces
6
 
7
- # preload model and tokenizer
8
- AutoModelForCausalLM.from_pretrained("Rorical/0-roleplay", return_dict=True, trust_remote_code=True)
9
  tokenizer = AutoTokenizer.from_pretrained("Rorical/0-roleplay", trust_remote_code=True)
10
  tokenizer.add_special_tokens({"bos_token": tokenizer.eos_token})
11
  tokenizer.bos_token_id = tokenizer.eos_token_id
 
4
 
5
  import spaces
6
 
7
+ # Load tokenizer
 
8
  tokenizer = AutoTokenizer.from_pretrained("Rorical/0-roleplay", trust_remote_code=True)
9
  tokenizer.add_special_tokens({"bos_token": tokenizer.eos_token})
10
  tokenizer.bos_token_id = tokenizer.eos_token_id