Add silent mode

README.md

@@ -301,8 +301,10 @@ You can set environment variables to control the meta-behavior of DiscoArt. The
 ```bash
 DISCOART_LOG_LEVEL='DEBUG' # more verbose logs
 DISCOART_OPTOUT_CLOUD_BACKUP='1' # opt-out from cloud backup
+DISCOART_OPTOUT_LOCAL_BACKUP='1' # opt-out from local backup
 DISCOART_DISABLE_IPYTHON='1' # disable ipython dependency
 DISCOART_DISABLE_RESULT_SUMMARY='1' # disable result summary after the run ends
+DISCOART_DISABLE_ARGS_TABLE='1' # disable args table before the run starts
 DISCOART_DEFAULT_PARAMETERS_YAML='path/to/your-default.yml' # use a custom default parameters file
 DISCOART_CUT_SCHEDULES_YAML='path/to/your-schedules.yml' # use a custom cut schedules file
 DISCOART_MODELS_YAML='path/to/your-models.yml' # use a custom list of models file

discoart/__init__.py

@@ -19,3 +19,4 @@
 
 from .create import create, go_big
 from .config import cheatsheet, show_config, save_config, load_config
+from .helper import get_output_dir

discoart/create.py

@@ -185,7 +185,9 @@ def create(**kwargs) -> Optional['DocumentArray']:
     else:
         _args = load_config(user_config=kwargs)
 
-    print_args_table(_args)
+    if 'DISCOART_DISABLE_ARGS_TABLE' not in os.environ:
+        print_args_table(_args)
+
     _args = SimpleNamespace(**_args)
 
     from .helper import (
@@ -220,10 +222,10 @@ def create(**kwargs) -> Optional['DocumentArray']:
             device=device,
             events=events,
         )
-        is_exit0 = True
+        is_exit0 = 'DISCOART_OPTOUT_LOCAL_BACKUP' not in os.environ
         return da
     except KeyboardInterrupt:
-        is_exit0 = True
+        is_exit0 = 'DISCOART_OPTOUT_LOCAL_BACKUP' not in os.environ
     finally:
         free_memory()
 

discoart/persist.py

@@ -144,6 +144,8 @@ def _local_save(
     is_sampling_done: threading.Event,
     force: bool = False,
 ) -> None:
+    if 'DISCOART_OPTOUT_LOCAL_BACKUP' in os.environ:
+        return
     if is_busy_event.is_set() and not force:
         logger.debug(f'another save is running, skipping')
         return

discoart/runner.py

@@ -45,7 +45,7 @@ def do_run(
     logger.info('preparing models...')
 
     model, diffusion, clip_models, secondary_model = models
-    lpips_model = lpips.LPIPS(net='vgg').to(device)
+    lpips_model = lpips.LPIPS(net='vgg', verbose=False).to(device)
 
     side_x, side_y = ((args.width_height[j] // 64) * 64 for j in (0, 1))