ffreemt commited on
Commit
9263f1a
1 Parent(s): 68482b0
Files changed (1) hide show
  1. app.py +9 -4
app.py CHANGED
@@ -1,4 +1,4 @@
1
- # pylint: disable=broad-exception-caught, redefined-outer-name, missing-function-docstring, missing-module-docstring, too-many-arguments, line-too-long, invalid-name, redefined-builtin
2
  # import gradio as gr
3
 
4
  # model_name = "models/THUDM/chatglm2-6b-int4"
@@ -7,10 +7,10 @@
7
  # %%writefile demo-4bit.py
8
 
9
  from textwrap import dedent
10
- import torch
11
 
12
  import gradio as gr
13
  import mdtex2html
 
14
  from loguru import logger
15
 
16
  # credit to https://github.com/THUDM/ChatGLM2-6B/blob/main/web_demo.py
@@ -99,7 +99,12 @@ def predict(
99
  chatbot.append((parse_text(input), ""))
100
  except Exception as exc:
101
  logger.error(exc)
102
- chatbot[-1] = (parse_text(input), str(exc))
 
 
 
 
 
103
  yield chatbot, history, past_key_values
104
 
105
  for response, history, past_key_values in model.stream_chat(
@@ -230,7 +235,7 @@ with gr.Blocks(title="ChatGLM2-6B-int4", theme=gr.themes.Soft(text_size="sm")) a
230
  max_length = gr.Slider(
231
  0,
232
  32768,
233
- value=8192 / 2,
234
  step=1.0,
235
  label="Maximum length",
236
  interactive=True,
 
1
+ # pylint: disable=broad-exception-caught, redefined-outer-name, missing-function-docstring, missing-module-docstring, too-many-arguments, line-too-long, invalid-name, redefined-builtin, redefined-argument-from-local
2
  # import gradio as gr
3
 
4
  # model_name = "models/THUDM/chatglm2-6b-int4"
 
7
  # %%writefile demo-4bit.py
8
 
9
  from textwrap import dedent
 
10
 
11
  import gradio as gr
12
  import mdtex2html
13
+ import torch
14
  from loguru import logger
15
 
16
  # credit to https://github.com/THUDM/ChatGLM2-6B/blob/main/web_demo.py
 
99
  chatbot.append((parse_text(input), ""))
100
  except Exception as exc:
101
  logger.error(exc)
102
+ logger.debug(f"{chatbot=}")
103
+ _ = """
104
+ if chatbot:
105
+ chatbot[-1] = (parse_text(input), str(exc))
106
+ yield chatbot, history, past_key_values
107
+ # """
108
  yield chatbot, history, past_key_values
109
 
110
  for response, history, past_key_values in model.stream_chat(
 
235
  max_length = gr.Slider(
236
  0,
237
  32768,
238
+ value=8192,
239
  step=1.0,
240
  label="Maximum length",
241
  interactive=True,