Skip to content

Commit 92b7bef

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 4fed793 commit 92b7bef

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

examples/controlnet/train_controlnet_flax.py

Lines changed: 0 additions & 4 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
@@ -1028,9 +1027,6 @@ def l2(xs):
10281027
}
10291028
)
10301029

1031-
if args.ccache:
1032-
cc.initialize_cache(args.ccache)
1033-
10341030
global_step = step0 = 0
10351031
epochs = tqdm(
10361032
range(args.num_train_epochs),

0 commit comments

Comments
 (0)