Upload folder using huggingface_hub
Browse files- inference.py +2 -2
inference.py
CHANGED
|
@@ -26,8 +26,8 @@ def inference(
|
|
| 26 |
content_image,
|
| 27 |
style_features,
|
| 28 |
lr,
|
| 29 |
-
iterations=
|
| 30 |
-
optim_caller=optim.
|
| 31 |
alpha=1,
|
| 32 |
beta=1
|
| 33 |
):
|
|
|
|
| 26 |
content_image,
|
| 27 |
style_features,
|
| 28 |
lr,
|
| 29 |
+
iterations=3,
|
| 30 |
+
optim_caller=optim.LBFGS,
|
| 31 |
alpha=1,
|
| 32 |
beta=1
|
| 33 |
):
|