Spaces:
Sleeping
Sleeping
Upload folder using huggingface_hub
Browse files- inference.py +1 -1
inference.py
CHANGED
@@ -33,7 +33,7 @@ def inference(
|
|
33 |
):
|
34 |
generated_image = content_image.clone().requires_grad_(True)
|
35 |
optimizer = optim_caller([generated_image], lr=lr)
|
36 |
-
min_losses = [
|
37 |
|
38 |
with torch.no_grad():
|
39 |
content_features = model(content_image)
|
|
|
33 |
):
|
34 |
generated_image = content_image.clone().requires_grad_(True)
|
35 |
optimizer = optim_caller([generated_image], lr=lr)
|
36 |
+
min_losses = [float('inf')] * iterations
|
37 |
|
38 |
with torch.no_grad():
|
39 |
content_features = model(content_image)
|