jamino30 commited on
Commit
cf753ac
·
verified ·
1 Parent(s): 0fc4e06

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. 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 tqdm(range(iterations), desc='The magic is happening ✨'):
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