Skip to content

Commit d413881

Browse files
authored
[Docs] Update gguf.md to remove generator from the pipeline from_pretrained (#10299)
Update gguf.md to remove generator from the pipeline from_pretrained
1 parent a6288a5 commit d413881

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

docs/source/en/quantization/gguf.md

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -45,12 +45,11 @@ transformer = FluxTransformer2DModel.from_single_file(
4545
pipe = FluxPipeline.from_pretrained(
4646
"black-forest-labs/FLUX.1-dev",
4747
transformer=transformer,
48-
generator=torch.manual_seed(0),
4948
torch_dtype=torch.bfloat16,
5049
)
5150
pipe.enable_model_cpu_offload()
5251
prompt = "A cat holding a sign that says hello world"
53-
image = pipe(prompt).images[0]
52+
image = pipe(prompt, generator=torch.manual_seed(0)).images[0]
5453
image.save("flux-gguf.png")
5554
```
5655

0 commit comments

Comments
 (0)