fix to use actual prompt (#227)

2025-09-01 04:14:38 +08:00 · 2024-01-04 11:12:05 -08:00
parent a9ee883bf7
commit 7e7b817ddf
1 changed files with 1 additions and 1 deletions
--- a/llms/hf_llm/generate.py
+++ b/llms/hf_llm/generate.py
@@ -16,7 +16,7 @@ def generate(
    temp: float = 0.0,
 ):
    prompt = tokenizer(
-        args.prompt,
+        prompt,
        return_tensors="np",
        return_attention_mask=False,
    )[