Shiningknob721 commited on
Commit
db25d60
·
verified ·
1 Parent(s): 3bad1f7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import pipeline
4
  tsundere_ai = pipeline(
5
  "text-generation",
6
  model="rinna/japanese-gpt-neox-3.6b-instruction-ppo",
7
- device_map=None
8
  tokenizer_kwargs={"use_fast": False}
9
  )
10
 
 
4
  tsundere_ai = pipeline(
5
  "text-generation",
6
  model="rinna/japanese-gpt-neox-3.6b-instruction-ppo",
7
+ device_map=None,
8
  tokenizer_kwargs={"use_fast": False}
9
  )
10