Flux LoRA Collections
Collection
Flux THE LoRA
β’
87 items
β’
Updated
β’
28
The model is still in the training phase. This is not the final version and may contain artifacts and perform poorly in some cases.
prithivMLmods/Castor-3D-Sketchfab-Flux-LoRA
Image Processing Parameters
Parameter | Value | Parameter | Value |
---|---|---|---|
LR Scheduler | constant | Noise Offset | 0.03 |
Optimizer | AdamW | Multires Noise Discount | 0.1 |
Network Dim | 64 | Multires Noise Iterations | 10 |
Network Alpha | 32 | Repeat & Steps | 23 & 1.8k |
Epoch | 10 | Save Every N Epochs | 1 |
Labeling: florence2-en(natural language & English)
Total Images Used for Training : 39
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)
/project-root/
βββ .gitattributes
βββ README.md
βββ app.py
βββ pythonproject.py
You should use 3D Sketchfab
to trigger the image generation.
Weights for this model are available in Safetensors format.
Download them in the Files & versions tab.
Base model
black-forest-labs/FLUX.1-dev