Spaces:
Build error
Build error
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
|