fix to use actual prompt

2025-09-01 04:14:38 +08:00 · 2024-01-04 11:07:36 -08:00
parent a5d6d0436c
commit caf314e703
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,
    )[