"src/graph/sampling/vscode:/vscode.git/clone" did not exist on "5fcd7f290fcb2e636e431e39c58e18d7efb42a08"
Unverified Commit d4138814 authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

[Docs] Update gguf.md to remove generator from the pipeline from_pretrained (#10299)

Update gguf.md to remove generator from the pipeline from_pretrained
parent a6288a55
...@@ -45,12 +45,11 @@ transformer = FluxTransformer2DModel.from_single_file( ...@@ -45,12 +45,11 @@ transformer = FluxTransformer2DModel.from_single_file(
pipe = FluxPipeline.from_pretrained( pipe = FluxPipeline.from_pretrained(
"black-forest-labs/FLUX.1-dev", "black-forest-labs/FLUX.1-dev",
transformer=transformer, transformer=transformer,
generator=torch.manual_seed(0),
torch_dtype=torch.bfloat16, torch_dtype=torch.bfloat16,
) )
pipe.enable_model_cpu_offload() pipe.enable_model_cpu_offload()
prompt = "A cat holding a sign that says hello world" prompt = "A cat holding a sign that says hello world"
image = pipe(prompt).images[0] image = pipe(prompt, generator=torch.manual_seed(0)).images[0]
image.save("flux-gguf.png") image.save("flux-gguf.png")
``` ```
......
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