fix web ui demo

bewithme · Nov 16, 2023 · 10ce87e · 10ce87e
1 parent 1c80e9a
commit 10ce87e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/llmtuner/webui/chatter.py b/src/llmtuner/webui/chatter.py
@@ -29,14 +29,14 @@ def __init__(
  if not lazy_init: # read arguments from command line
  super().__init__()
 
- if demo_mode: # load config.json by default
+ if demo_mode: # load demo_config.json if exists
  import json
  try:
  with open("demo_config.json", "r", encoding="utf-8") as f:
  args = json.load(f)
+ super().__init__(args)
  except:
- raise ValueError("Cannot find `demo_config.json` at current directory.")
- super().__init__(args)
+ print("Cannot find `demo_config.json` at current directory.")
 
  @property
  def loaded(self) -> bool: