Jialuo21/Science-T2I-Flux-SFT

Resources

• Website
• arXiv: Paper
• GitHub: Code
• Huggingface: SciScore
• Huggingface: Science-T2I-S&C Benchmark
• Huggingface: Science-T2I training set

Quick Start

You can use FluxPipeline to run the model

import torch
from diffusers import FluxPipeline
pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16).to('cuda')
pipe.load_lora_weights("Jialuo21/Science-T2I-Flux-SFT")

prompt = "A football in a space without gravity"
image = pipe(
    prompt,
    height=1024,
    width=1024,
    guidance_scale=3.5,
    num_inference_steps=50,
    max_sequence_length=512,
    generator=torch.Generator("cpu").manual_seed(0)
).images[0]
image.save("example.png")