Spaces:

Tonic
/

RWKV-7

Running

Tonic commited on Jan 8

Commit

bcc02ff

verified ·

1 Parent(s): 2a7a96e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -61,8 +61,7 @@ class ModelManager:
             download_model(MODELS[model_name])
             self.current_model = RWKV(
                 model=MODELS[model_name],
-                strategy='cpu fp32',
-                v7=True
             )
             self.pipeline = PIPELINE(self.current_model, TOKENIZER_FILE)
             self.current_model_name = model_name

             download_model(MODELS[model_name])
             self.current_model = RWKV(
                 model=MODELS[model_name],
+                strategy='cpu fp32'
             )
             self.pipeline = PIPELINE(self.current_model, TOKENIZER_FILE)
             self.current_model_name = model_name