prithivMLmods/Castor-3D-Sketchfab-Flux-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("prithivMLmods/Castor-3D-Sketchfab-Flux-LoRA")

prompt = "3D Sketchfab, A low-angle view of a brown military tank on a bright yellow background. The tank has a red number \"88\" on the side of it. There is a black X on the top of the tank. There are two black wheels on the bottom of it and a black stripe on the right side of the front of the left side of this tank."
image = pipe(prompt).images[0]

Configuración alternativa mostrada en la tarjeta:

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 = "prithivMLmods/Castor-3D-Sketchfab-Flux-LoRA"
trigger_word = "3D Sketchfab"  # Leave trigger_word blank if not used.
pipe.load_lora_weights(lora_repo)

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