strangerzonehf/Flux-Isometric-3D-LoRA

Instalación y uso con Diffusers:

pip install -U diffusers transformers accelerate

import torch
from diffusers import DiffusionPipeline

# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained(
    "black-forest-labs/FLUX.1-dev",
    dtype=torch.bfloat16,
    device_map="cuda"
)
pipe.load_lora_weights("strangerzonehf/Flux-Isometric-3D-LoRA")

prompt = "Isometric 3D, a 3D model of a tropical island is displayed on a light blue backdrop. The island features a small body of water, surrounded by gray rocks and green grass. There are palm trees and small bushes scattered throughout the island, adding a pop of color to the scene."
image = pipe(prompt).images[0]

Configuración alternativa indicada en la tarjeta del modelo:

import torch
from pipelines import DiffusionPipeline

base_model = "black-forest-labs/FLUX.1-dev"
pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)

lora_repo = "strangerzonehf/Flux-Isometric-3D-LoRA"
trigger_word = "Isometric 3D"
pipe.load_lora_weights(lora_repo)

device = torch.device("cuda")
pipe.to(device)