Spaces:

data-agents
/

jupyter-agent

Running

lvwerra HF staff commited on 25 days ago

Commit

6b6ff22

verified ·

1 Parent(s): fa34d67

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -220,7 +220,7 @@ def run_interactive_notebook(client, model, tokenizer, messages, sbx, max_new_to
         while True:
             response_stream = client.text_generation(
                 model=model,
-                messages=model_input,
                 details=True,
                 stream=True,
                 max_tokens=max_new_tokens,

         while True:
             response_stream = client.text_generation(
                 model=model,
+                prompt=model_input,
                 details=True,
                 stream=True,
                 max_tokens=max_new_tokens,