Views
No views yet
1import torch
2from diffusers import FluxPipeline
3
4pipe = FluxPipeline.from_pretrained("tlwu/tiny-random-flux", torch_dtype=torch.bfloat16)
5prompt = "a tree with blue leaves"
6image = pipe(
7 prompt,
8 height=1024,
9 width=1024,
10 guidance_scale=3.5,
11 num_inference_steps=50,
12 max_sequence_length=512,
13 generator=torch.Generator("cpu").manual_seed(0)
14).images[0]
15
16image.save("test.png")