custom launch

launch.py

@@ -97,6 +97,7 @@ if config.temp_path_cleanup_on_launch:
 
 
 def download_models(default_model, previous_default_models, checkpoint_downloads, embeddings_downloads, lora_downloads):
+    print('DOWNLOADING MODELS...')
     for file_name, url in vae_approx_filenames:
         load_file_from_url(url=url, model_dir=config.path_vae_approx, file_name=file_name)
 
@@ -122,18 +123,19 @@ def download_models(default_model, previous_default_models, checkpoint_downloads
                     default_model = alternative_model_name
                     break
 
-    print('checkpoint_downloads=', checkpoint_downloads)
+    print('CHECKPOINT_DOWNLOADS=', checkpoint_downloads)
     for file_name, url in checkpoint_downloads.items():
         load_file_from_url(url=url, model_dir=config.paths_checkpoints[0], file_name=file_name)
         
-    print('embeddings_downloads=', embeddings_downloads)
+    print('EMBEDDINGS_DOWNLOADS=', embeddings_downloads)
     for file_name, url in embeddings_downloads.items():
         load_file_from_url(url=url, model_dir=config.path_embeddings, file_name=file_name)
 
-    print('lora_downloads=', lora_downloads)
+    print('LORA_DOWNLOADS=', lora_downloads)
     for file_name, url in lora_downloads.items():
         load_file_from_url(url=url, model_dir=config.paths_loras[0], file_name=file_name)
 
+    print('MODELS DOWNLOADED.')
     return default_model, checkpoint_downloads