Spaces:
Sleeping
Sleeping
Commit
·
607417a
1
Parent(s):
5fb8402
Update app.py
Browse files
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(
|
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
|