Skip to content

GitLab

Unverified Commit eb29dbad authored by Jair Trejo's avatar Jair Trejo Committed by GitHub
Browse files

Fix typo in textual inversion JAX training script (#3123) 

The pipeline is built as `pipe` but then used as `pipeline`.
parent d92c4d5a
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
docs/source/en/training/text_inversion.mdx
View file @ eb29dbad
...@@ -245,7 +245,7 @@ from flax.training.common_utils import shard ...@@ -245,7 +245,7 @@ from flax.training.common_utils import shard
from diffusers import FlaxStableDiffusionPipeline from diffusers import FlaxStableDiffusionPipeline
model_path = "path-to-your-trained-model" model_path = "path-to-your-trained-model"
pipe, params = FlaxStableDiffusionPipeline.from_pretrained(model_path, dtype=jax.numpy.bfloat16) pipeline, params = FlaxStableDiffusionPipeline.from_pretrained(model_path, dtype=jax.numpy.bfloat16)
prompt = "A <cat-toy> backpack" prompt = "A <cat-toy> backpack"
prng_seed = jax.random.PRNGKey(0) prng_seed = jax.random.PRNGKey(0)
......
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