I trained this model using the Diffusers library by randomly selecting layers and blocks (not training every layer), which reduced the training time and is expected to yield better results.
import torch
from diffusers import FluxPipeline
pipe = FluxPipeline.from_pretrained("kpsss34/FHDR_Uncensored", torch_dtype=torch.bfloat16)
pipe.enable_model_cpu_offload()
prompt = "a women..."
image = pipe(
prompt,
height=1024,
width=1024,
guidance_scale=4.0,
num_inference_steps=40,
max_sequence_length=512,
generator=torch.Generator("cpu").manual_seed(0)
).images[0]
image.save("outputs.png")
- Downloads last month
- 104
Hardware compatibility
Log In
to view the estimation
8-bit
Model tree for kpsss34/FHDR_Uncensored
Base model
black-forest-labs/FLUX.1-dev

