flux python starter files

flux.py

import torch
from diffusers import FluxPipeline

prompt = "A cat holding a sign that says hello world"

model = "/home/yakov/.cache/huggingface/hub/models--black-forest-labs--FLUX.1-schnell/snapshots/741f7c3ce8b383c54771c7003378a50191e9efe9" 
device = "cuda"

pipe = FluxPipeline.from_pretrained(model, torch_dtype=torch.bfloat16).to(device)
pipe.enable_model_cpu_offload()

image = pipe(
    prompt,
    guidance_scale=0.0,
    num_inference_steps=4,
    max_sequence_length=256,
    generator = torch.Generator(device=device).manual_seed(42)
).images[0]
image.save("flux-schnell.png")

requirements.txt

accelerate
diffusers
torch
transformers
transformers[sentencepiece]

venv-activate.sh

#! /bin/bash

source ./venv/bin/activate

venv-create.sh

#! /bin/bash

python -m venv venv