๐ Castor-3D-Sketchfab-Flux-LoRA
This model focuses on text - to - image generation, leveraging LoRA and Diffusers. It's currently in the training phase, aiming to generate high - quality 3D Sketchfab - style images.
๐ Model Tags
- text - to - image
- lora
- diffusers
- template:diffusion - lora
๐ผ๏ธ Model Widget Examples
- Input Text 1: '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.'
- Input Text 2: '3D Sketchfab, An eye - level view of a small red building with a green sign on the front of it. The building is surrounded by a small patch of brown dirt. There is a brown roof on the right side of the building and a red door on the left side. There are small green plants on the ground around the building. A black telephone pole is in the center of the image. A white dish is on the roof of the house.'
- Input Text 3: '3D Sketchfab, a vibrant orange box is adorned with a fish sculpture. The fish sculpture is a vibrant shade of blue, with a striped pattern on its body. It is positioned on a light purple background, with the fish head facing towards the right side of the frame. A knife with a black handle is positioned to the left of the fish sculpture, positioned on top of the box.'
๐ Quick Start
๐ ๏ธ Setting Up
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"
pipe.load_lora_weights(lora_repo)
device = torch.device("cuda")
pipe.to(device)
โ ๏ธ Important Note
The model is still in the training phase. This is not the final version and may contain artifacts and perform poorly in some cases.
โจ Features
๐ Model description
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
๐ App File Structure
/project-root/
โโโ .gitattributes
โโโ README.md
โโโ app.py
โโโ pythonproject.py
๐งจ Trigger words
You should use 3D Sketchfab
to trigger the image generation.
๐ฆ Installation
The weights for this model are available in Safetensors format. You can Download them in the Files & versions tab.
๐ License
This model is licensed under the creativeml - openrail - m
license.