Spaces:
Running
on
Zero
Running
on
Zero
Upload folder using huggingface_hub
Browse files- inference.py +3 -3
inference.py
CHANGED
@@ -75,7 +75,7 @@ def inference(
|
|
75 |
min_losses[iter] = min(min_losses[iter], total_loss.item())
|
76 |
return total_loss
|
77 |
|
78 |
-
for iter in
|
79 |
optimizer.step(lambda: closure(iter))
|
80 |
|
81 |
if apply_to_background:
|
@@ -84,5 +84,5 @@ def inference(
|
|
84 |
generated_image.data = generated_image.data * (1 - foreground_mask_resized) + content_image.data * foreground_mask_resized
|
85 |
|
86 |
if iter % 10 == 0: print(f'Loss ({iter}):', min_losses[iter])
|
87 |
-
|
88 |
-
return generated_image
|
|
|
75 |
min_losses[iter] = min(min_losses[iter], total_loss.item())
|
76 |
return total_loss
|
77 |
|
78 |
+
for iter in range(iterations):
|
79 |
optimizer.step(lambda: closure(iter))
|
80 |
|
81 |
if apply_to_background:
|
|
|
84 |
generated_image.data = generated_image.data * (1 - foreground_mask_resized) + content_image.data * foreground_mask_resized
|
85 |
|
86 |
if iter % 10 == 0: print(f'Loss ({iter}):', min_losses[iter])
|
87 |
+
|
88 |
+
return generated_image
|