jchen8000 commited on
Commit
6eec9cd
·
1 Parent(s): fb1fbd9

update server.py

Browse files
Files changed (1) hide show
  1. server.py +2 -0
server.py CHANGED
@@ -175,6 +175,7 @@ def create_interface():
175
  if __name__ == "__main__":
176
 
177
  logger.info("Starting Text generation web UI")
 
178
  do_cmd_flags_warnings()
179
 
180
  # Load custom settings
@@ -208,6 +209,7 @@ if __name__ == "__main__":
208
  # Model defined through --model
209
  if shared.args.model is not None:
210
  shared.model_name = shared.args.model
 
211
 
212
  # Select the model from a command-line menu
213
  elif shared.args.model_menu:
 
175
  if __name__ == "__main__":
176
 
177
  logger.info("Starting Text generation web UI")
178
+ logger.info("Need to download models...")
179
  do_cmd_flags_warnings()
180
 
181
  # Load custom settings
 
209
  # Model defined through --model
210
  if shared.args.model is not None:
211
  shared.model_name = shared.args.model
212
+ logger.info("Shared model:" + shared.model_name)
213
 
214
  # Select the model from a command-line menu
215
  elif shared.args.model_menu: