Castor 3D Portrait Flux LoRA

🚀 Castor-3D-Portrait-Flux-LoRA

Castor-3D-Portrait-Flux-LoRA 是一个用于图像生成的模型，可基于给定的文本提示生成 3D 肖像风格的卡通图像。不过该模型仍处于训练阶段，当前版本可能存在瑕疵，在某些情况下表现不佳。

🚀 快速开始

模型设置

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-Portrait-Flux-LoRA"
trigger_word = "3D Portrait"  # 若不使用触发词，可留空。
pipe.load_lora_weights(lora_repo)

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

✨ 主要特性

• 能够根据文本提示生成 3D 肖像风格的卡通图像。
• 支持使用触发词 3D Portrait 来触发图像生成。

📦 安装指南

暂未提供具体的安装步骤，可参考上述代码示例进行模型的加载和设置。

💻 使用示例

基础用法

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-Portrait-Flux-LoRA"
trigger_word = "3D Portrait"  # 若不使用触发词，可留空。
pipe.load_lora_weights(lora_repo)

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

# 此处可添加生成图像的代码

📚 详细文档

模型描述

prithivMLmods/Castor-3D-Portrait-Flux-LoRA

图像参数处理

标注：florence2-en（自然语言 & 英语）

训练使用的总图像数：19

应用文件结构

/project-root/

├── .gitattributes
├── README.md
├── app.py
├── pythonproject.py

触发词 🧨

⚠️ 重要提示

你应该使用 3D Portrait 来触发图像生成。

下载模型

此模型的权重以 Safetensors 格式提供。

点击下载，可在“Files & versions” 标签中获取。

📄 许可证

本模型使用的许可证为 creativeml-openrail-m。