Update app.py

app.py

@@ -311,8 +311,8 @@ def inference(text, seed, step_size, max_iterations, width, height, init_image,
         
         result = []
         if init_weight:
-            # result.append(F.mse_loss(z, z_orig) * init_weight / 2)
-            result.append(F.mse_loss(z, torch.zeros_like(z_orig)) * ((1/torch.tensor(i*2 + 1))*init_weight) / 2)
+            result.append(F.mse_loss(z, z_orig) * init_weight / 2)
+            #result.append(F.mse_loss(z, torch.zeros_like(z_orig)) * ((1/torch.tensor(i*2 + 1))*init_weight) / 2)
         for prompt in pMs:
             result.append(prompt(iii))
         img = np.array(out.mul(255).clamp(0, 255)[0].cpu().detach().numpy().astype(np.uint8))[:,:,:]