🚀 Castor-3D-Sketchfab-Flux-LoRA
本模型是一个用于图像生成的模型,基于特定的基础模型进行训练,可通过特定触发词生成相关的3D风格图像。目前该模型仍处于训练阶段,最终版本可能会有更好的表现。
🚀 快速开始
模型描述
prithivMLmods/Castor-3D-Sketchfab-Flux-LoRA
图像生成参数如下:
参数 |
值 |
参数 |
值 |
LR调度器 |
constant |
噪声偏移 |
0.03 |
优化器 |
AdamW |
多分辨率噪声折扣 |
0.1 |
网络维度 |
64 |
多分辨率噪声迭代次数 |
10 |
网络Alpha |
32 |
重复次数与步数 |
23 & 1.8k |
训练轮数 |
10 |
每N轮保存一次 |
1 |
标注方式:florence2 - en(自然语言 & 英语)
训练使用的总图像数: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"
pipe.load_lora_weights(lora_repo)
device = torch.device("cuda")
pipe.to(device)
应用文件结构
/project-root/
├── .gitattributes
├── README.md
├── app.py
├── pythonproject.py
触发词 🧨
你应该使用 3D Sketchfab
来触发图像生成。
下载模型
该模型的权重以Safetensors格式提供。
在文件与版本标签中下载。
注意事项
⚠️ 重要提示
该模型仍处于训练阶段,这不是最终版本,可能会包含瑕疵,并且在某些情况下表现不佳。
示例展示
以下是一些示例输入和对应的输出图像:
- 输入:'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.'
输出:图像链接
- 输入:'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.'
输出:图像链接
- 输入:'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.'
输出:图像链接
许可证
本模型使用的许可证为:creativeml - openrail - m。