Spaces:
Running
on
Zero
Running
on
Zero
Upload folder using huggingface_hub
Browse files- 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
|