picocreator commited on
Commit
607417a
1 Parent(s): 5fb8402

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ pipeline = PIPELINE(model, "rwkv_vocab_v20230424")
43
  def precompute_state(text):
44
  state = None
45
  text_encoded = pipeline.encode(text)
46
- _, state = model.forward(text_encodedm state)
47
  yield dict(state)
48
 
49
  # Precomputing the base instruction set
 
43
  def precompute_state(text):
44
  state = None
45
  text_encoded = pipeline.encode(text)
46
+ _, state = model.forward(text_encoded, state)
47
  yield dict(state)
48
 
49
  # Precomputing the base instruction set