Update llava/serve/gradio_web_server.py

llava/serve/gradio_web_server.py

@@ -238,7 +238,7 @@ def http_bot(state, model_selector, temperature, top_p, max_new_tokens, repeat_t
     model_name = model_selector
     
     if state.skip_next:
-        yield (state, state.to_gradio_chatbot(), None) + (no_change_btn,) * 2
+        yield (state, state.to_gradio_chatbot(), None) + (no_change_btn,) * 3
         return
 
     if len(state.messages) == state.offset + 2:
@@ -258,7 +258,7 @@ def http_bot(state, model_selector, temperature, top_p, max_new_tokens, repeat_t
     # No available worker
     if worker_addr == "":
         state.messages[-1][-1] = server_error_msg
-        yield (state, state.to_gradio_chatbot(), None, disable_btn, disable_btn)
+        yield (state, state.to_gradio_chatbot(), None, disable_btn, disable_btn, disable_btn)
         return
 
     # Construct prompt