Skip to content

Commit 0b70b28

Browse files
yiyixuxuandsteing
authored andcommitted
minor fix in controlnet flax example (huggingface#2986)
* fix the error when push_to_hub but not log validation * contronet_from_pt & controlnet_revision * add intermediate checkpointing to the guide
1 parent 366ad89 commit 0b70b28

File tree

2 files changed

+12
-0
lines changed

2 files changed

+12
-0
lines changed

examples/controlnet/README.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -326,6 +326,7 @@ If you want to use Weights and Biases logging, you should also install `wandb` n
326326
pip install wandb
327327
```
328328

329+
329330
Now let's downloading two conditioning images that we will use to run validation during the training in order to track our progress
330331

331332
```

examples/controlnet/train_controlnet_flax.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -216,6 +216,17 @@ def parse_args():
216216
action="store_true",
217217
help="Load the pretrained model from a PyTorch checkpoint.",
218218
)
219+
parser.add_argument(
220+
"--controlnet_revision",
221+
type=str,
222+
default=None,
223+
help="Revision of controlnet model identifier from huggingface.co/models.",
224+
)
225+
parser.add_argument(
226+
"--controlnet_from_pt",
227+
action="store_true",
228+
help="Load the controlnet model from a PyTorch checkpoint.",
229+
)
219230
parser.add_argument(
220231
"--profile_steps",
221232
type=int,

0 commit comments

Comments
 (0)