ttoosi commited on
Commit
0657874
·
verified ·
1 Parent(s): ff3c24a

Update app.py

Browse files

fixing grad minor bug

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -121,7 +121,7 @@ def simple_generative_inference(image, mode, model, n_iterations=10, step_size=0
121
  image_tensor = image_tensor + step_size * grad
122
  delta = image_tensor - image_original
123
  delta = torch.clamp(delta, -eps, eps) # Keep within range
124
- image_tensor = torch.clamp(image_original + diff, 0, 1)
125
  image_tensor = image_tensor.clone().detach().requires_grad_(True) # Ensure it's a new leaf tensor
126
  image_tensor.retain_grad()
127
 
 
121
  image_tensor = image_tensor + step_size * grad
122
  delta = image_tensor - image_original
123
  delta = torch.clamp(delta, -eps, eps) # Keep within range
124
+ image_tensor = torch.clamp(image_original + delta, 0, 1)
125
  image_tensor = image_tensor.clone().detach().requires_grad_(True) # Ensure it's a new leaf tensor
126
  image_tensor.retain_grad()
127