Skip to content

GitLab

Unverified Commit 1bd4c9e9 authored by YiYi Xu's avatar YiYi Xu Committed by GitHub
Browse files

remvoe one line as requested by gc team (#3077) 

remvoe one line
parent eb2ef316
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 4 deletions
examples/text_to_image/train_text_to_image_flax.py
View file @ 1bd4c9e9
...@@ -340,7 +340,6 @@ def main(): ...@@ -340,7 +340,6 @@ def main():
return examples return examples
if jax.process_index() == 0:
if args.max_train_samples is not None: if args.max_train_samples is not None:
dataset["train"] = dataset["train"].shuffle(seed=args.seed).select(range(args.max_train_samples)) dataset["train"] = dataset["train"].shuffle(seed=args.seed).select(range(args.max_train_samples))
# Set the training transforms # Set the training transforms
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment