jamino30 commited on
Commit
fa762f9
1 Parent(s): aa791f1

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -47,7 +47,7 @@ def inference(
47
  return total_loss
48
 
49
  for iter in tqdm(range(iterations), desc='The magic is happening ✨'):
50
- optimizer.step(closure)
51
  print(f'Loss ({iter+1}):', min_losses[iter])
52
 
53
  return generated_image
 
47
  return total_loss
48
 
49
  for iter in tqdm(range(iterations), desc='The magic is happening ✨'):
50
+ optimizer.step(lambda: closure(iter))
51
  print(f'Loss ({iter+1}):', min_losses[iter])
52
 
53
  return generated_image