NeurixYUFI/Neurix-Mega

Instalación y uso con Diffusers:

pip install -U diffusers transformers accelerate

import torch
from diffusers import DiffusionPipeline

# Cambia a "mps" para dispositivos Apple
pipe = DiffusionPipeline.from_pretrained(
    "black-forest-labs/FLUX.1-schnell",
    dtype=torch.bfloat16,
    device_map="cuda"
)
pipe.load_lora_weights("NeurixYUFI/Neurix-Mega")

prompt = "Cat on bed"
image = pipe(prompt).images[0]

Ejemplo alternativo del model card:

pip install diffusers transformers accelerate

from diffusers import DiffusionPipeline

pipeline = DiffusionPipeline.from_pretrained("NeurixYUFI/Neurix-Mega")
pipeline.to("cuda")  # si hay GPU disponible

prompt = "A vibrant landscape painting in the style of Van Gogh"
image = pipeline(prompt).images[0]
image.save("generated_image.png")

Parámetros mencionados para ajustar la generación:

image = pipeline(
    prompt,
    num_inference_steps=50,
    guidance_scale=7.5,
    seed=42
).images[0]