Skip to content

Commit eb29dba

Browse files
authored
Fix typo in textual inversion JAX training script (#3123)
The pipeline is built as `pipe` but then used as `pipeline`.
1 parent d92c4d5 commit eb29dba

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

docs/source/en/training/text_inversion.mdx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -245,7 +245,7 @@ from flax.training.common_utils import shard
245245
from diffusers import FlaxStableDiffusionPipeline
246246

247247
model_path = "path-to-your-trained-model"
248-
pipe, params = FlaxStableDiffusionPipeline.from_pretrained(model_path, dtype=jax.numpy.bfloat16)
248+
pipeline, params = FlaxStableDiffusionPipeline.from_pretrained(model_path, dtype=jax.numpy.bfloat16)
249249

250250
prompt = "A <cat-toy> backpack"
251251
prng_seed = jax.random.PRNGKey(0)

0 commit comments

Comments
 (0)