quickfix

app.py

@@ -96,7 +96,7 @@ def edit(
     elif model_id == LDM2_LARGE:
         ldm_stable = ldm2_large
     else:  # MUSIC
-    ldm_stable = ldm2_music
+        ldm_stable = ldm2_music
 
     ldm_stable.model.scheduler.set_timesteps(steps, device=device)
 
@@ -113,7 +113,7 @@ def edit(
             # do_inversion = True
             # Too much time has passed
     if wts is None or zs is None:
-            do_inversion = True
+        do_inversion = True
 
     if do_inversion or randomize_seed:  # always re-run inversion
         zs_tensor, wts_tensor = invert(ldm_stable=ldm_stable, x0=x0, prompt_src=source_prompt,
@@ -131,10 +131,10 @@ def edit(
         saved_inv_model = model_id
         do_inversion = False
     else:
-    #     wtszs = torch.load(wtszs_file, map_location=device)
-    #     # wtszs = torch.load(wtszs_file.f, map_location=device)
-    #     wts_tensor = wtszs['wts']
-    #     zs_tensor = wtszs['zs']
+        #     wtszs = torch.load(wtszs_file, map_location=device)
+        #     # wtszs = torch.load(wtszs_file.f, map_location=device)
+        #     wts_tensor = wtszs['wts']
+        #     zs_tensor = wtszs['zs']
         wts_tensor = wts.to(device)
         zs_tensor = zs.to(device)
 
@@ -224,7 +224,7 @@ change <code style="display:inline; background-color: lightgrey; ">duration = mi
 
 """
 
-with gr.Blocks(css='style.css') as demo: #, delete_cache=(3600, 3600)) as demo:
+with gr.Blocks(css='style.css') as demo:  #, delete_cache=(3600, 3600)) as demo:
     def reset_do_inversion(do_inversion_user, do_inversion):
         # do_inversion = gr.State(value=True)
         do_inversion = True
@@ -235,13 +235,13 @@ with gr.Blocks(css='style.css') as demo: #, delete_cache=(3600, 3600)) as demo:
     def clear_do_inversion_user(do_inversion_user):
         do_inversion_user = False
         return do_inversion_user
-    
+
     def post_match_do_inversion(do_inversion_user, do_inversion):
         if do_inversion_user:
             do_inversion = True
             do_inversion_user = False
         return do_inversion_user, do_inversion
-        
+
     gr.HTML(intro)
     wts = gr.State()
     zs = gr.State()
@@ -329,7 +329,7 @@ with gr.Blocks(css='style.css') as demo: #, delete_cache=(3600, 3600)) as demo:
                     saved_inv_model, do_inversion]  # , current_loaded_model, ldm_stable],
         ).then(post_match_do_inversion, inputs=[do_inversion_user, do_inversion], outputs=[do_inversion_user, do_inversion]
                ).then(lambda x: (demo.temp_file_sets.append(set([str(gr.utils.abspath(x))])) if type(x) is str else None),
-               inputs=wtszs)
+                      inputs=wtszs)
 
     # demo.move_resource_to_block_cache(wtszs.value)