Skip to content

Commit 2155c7b

Browse files
jairtrejoJimmy
authored andcommitted
Fix typo in textual inversion JAX training script (huggingface#3123)
The pipeline is built as `pipe` but then used as `pipeline`.
1 parent f302bd4 commit 2155c7b

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)