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。