svjack commited on
Commit
2adefaf
·
verified ·
1 Parent(s): 62be1bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -17,7 +17,8 @@ from cogvideox.ui.wan_fun_ui import ui, ui_eas, ui_modelscope
17
 
18
  if __name__ == "__main__":
19
  # Choose the ui mode
20
- ui_mode = "eas"
 
21
 
22
  # GPU memory mode, which can be choosen in [model_cpu_offload, model_cpu_offload_and_qfloat8, sequential_cpu_offload].
23
  # model_cpu_offload means that the entire model will be moved to the CPU after use, which can save some GPU memory.
@@ -56,7 +57,8 @@ if __name__ == "__main__":
56
  app, _, _ = demo.queue(status_update_rate=1).launch(
57
  server_name=server_name,
58
  server_port=server_port,
59
- prevent_thread_lock=True
 
60
  )
61
 
62
  # launch api
 
17
 
18
  if __name__ == "__main__":
19
  # Choose the ui mode
20
+ #ui_mode = "eas"
21
+ ui_mode = "local"
22
 
23
  # GPU memory mode, which can be choosen in [model_cpu_offload, model_cpu_offload_and_qfloat8, sequential_cpu_offload].
24
  # model_cpu_offload means that the entire model will be moved to the CPU after use, which can save some GPU memory.
 
57
  app, _, _ = demo.queue(status_update_rate=1).launch(
58
  server_name=server_name,
59
  server_port=server_port,
60
+ prevent_thread_lock=True,
61
+ share = True
62
  )
63
 
64
  # launch api