vlm-playground

Runtime error

lewtun HF staff commited on Apr 5

Commit

bbb29c3

•

1 Parent(s): 0fab3cb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -43,7 +43,7 @@ def bot_streaming(message, history):
   thread = Thread(target=model.generate, kwargs=generation_kwargs)
   thread.start()
-  text_prompt =f"USER: \n{message['text']}\nASSISTANT:" #f"[INST]  \n{message['text']} [/INST]"
   buffer = ""

   thread = Thread(target=model.generate, kwargs=generation_kwargs)
   thread.start()
+  text_prompt =f"USER:  \n{message['text']}\nASSISTANT: " #f"[INST]  \n{message['text']} [/INST]"
   buffer = ""