Spaces:

RWKV-Red-Team
/

RWKV-LatestSpace

Running on T4

sparkleman commited on Mar 7

Commit

0de6f92

1 Parent(s): 96dfb3e

FIX: typo

Files changed (1) hide show

app.py CHANGED Viewed

@@ -81,7 +81,7 @@ for model_config in CONFIG.MODELS:
     logger.info(f"Load Model - Path - {model_config.MODEL_FILE_PATH}")
     tmp_model = RWKV(
-        model=model_config.DOWNLOAD_MODEL_FILE_NAME.replace(".pth", ""),
         strategy=CONFIG.STRATEGY,
     )
     tmp_pipeline = PIPELINE(tmp_model, model_config.VOCAB)

     logger.info(f"Load Model - Path - {model_config.MODEL_FILE_PATH}")
     tmp_model = RWKV(
+        model=model_config.MODEL_FILE_PATH.replace(".pth", ""),
         strategy=CONFIG.STRATEGY,
     )
     tmp_pipeline = PIPELINE(tmp_model, model_config.VOCAB)