Skip to content

Commit 571ff47

Browse files
committed
Removes --ccache.
This makes only a very small difference (~1 min) with this model size, so removing the option introduced in cdb3cc.
1 parent c8d4dec commit 571ff47

File tree

1 file changed

+0
-10
lines changed

1 file changed

+0
-10
lines changed

examples/controlnet/train_controlnet_flax.py

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,6 @@
2222
from pathlib import Path
2323

2424
import jax
25-
import jax.experimental.compilation_cache.compilation_cache as cc
2625
import jax.numpy as jnp
2726
import numpy as np
2827
import optax
@@ -243,12 +242,6 @@ def parse_args():
243242
action="store_true",
244243
help="Whether to dump an initial (before training loop) and a final (at program end) memory profile.",
245244
)
246-
parser.add_argument(
247-
"--ccache",
248-
type=str,
249-
default=None,
250-
help="Enables compilation cache.",
251-
)
252245
parser.add_argument(
253246
"--tokenizer_name",
254247
type=str,
@@ -1017,9 +1010,6 @@ def l2(xs):
10171010
}
10181011
)
10191012

1020-
if args.ccache:
1021-
cc.initialize_cache(args.ccache)
1022-
10231013
global_step = step0 = 0
10241014
epochs = tqdm(
10251015
range(args.num_train_epochs),

0 commit comments

Comments
 (0)