jamino30 commited on
Commit
c5d8238
1 Parent(s): 5e1fe76

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. 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 = [[float('inf')]] * iterations
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)