Spaces:
Build error
Build error
Update app.py
Browse filesfixing grad minor bug
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 +
|
| 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 |
|