update space

app.py

@@ -272,7 +272,7 @@ def main():
         model.load_state_dict(ckp['model'])
 
         noise_initial = initial_noise[e_type].to(device)
-        noise_start_t = noise_start_t[e_type]
+        noise_start_t_e_type = noise_start_t[e_type]
         general_save_path = './logs/edited'
         os.makedirs(general_save_path, exist_ok=True)
         for i, latent in enumerate(state['latent']):
@@ -280,7 +280,7 @@ def main():
             text_embeddings_clip = model.cached_model_kwargs(1, dict(texts=[instruction]))
             print("shape of latent: ", latent.clone().unsqueeze(0).shape, "instruction: ", instruction)
             ref_latent = latent.clone().unsqueeze(0)
-            t_1 = torch.randint(noise_start_t, noise_start_t + 1, (1,), device=device).long()
+            t_1 = torch.randint(noise_start_t_e_type, noise_start_t_e_type + 1, (1,), device=device).long()
 
             noise_input = diffusion.q_sample(ref_latent, t_1, noise=noise_initial)
             out_1 = diffusion.p_mean_variance(model, noise_input, t_1, clip_denoised=True,