BRIA 3.1

🚀 BRIA 3.1文本到圖像模型

BRIA 3.1是一款全新的文本到圖像模型，它僅使用完全授權的數據進行訓練，能夠實現高質量的圖像生成。該模型為開發者提供API訪問和模型權重直接訪問兩種方式，方便集成。它擁有40億參數，相對輕量級，同時具備高視覺保真度和強大的提示對齊能力。

🚀 快速開始

BRIA 3.1模型可通過API、ComfyUI節點或直接獲取模型權重的方式使用。若需獲取模型權重，需購買商業許可證。以下是使用Diffusers庫調用模型的示例代碼：

pip install diffusers, hf_hub_download

from huggingface_hub import hf_hub_download
import os

try:
    local_dir = os.path.dirname(__file__)
except:
    local_dir = '.'
    
hf_hub_download(repo_id="briaai/BRIA-3.1", filename='pipeline_bria.py', local_dir=local_dir)
hf_hub_download(repo_id="briaai/BRIA-3.1", filename='transformer_bria.py', local_dir=local_dir)
hf_hub_download(repo_id="briaai/BRIA-3.1", filename='bria_utils.py', local_dir=local_dir)

import torch
from pipeline_bria import BriaPipeline

# trust_remote_code = True - allows loading a transformer which is not present at the transformers library(from transformer/bria_transformer.py)
pipe = BriaPipeline.from_pretrained("briaai/BRIA-3.1", torch_dtype=torch.bfloat16,trust_remote_code=True)
pipe.to(device="cuda")

prompt = "A portrait of a Beautiful and playful ethereal singer, golden designs, highly detailed, blurry background"
negative_prompt = "Logo,Watermark,Ugly,Morbid,Extra fingers,Poorly drawn hands,Mutation,Blurry,Extra limbs,Gross proportions,Missing arms,Mutated hands,Long neck,Duplicate,Mutilated,Mutilated hands,Poorly drawn face,Deformed,Bad anatomy,Cloned face,Malformed limbs,Missing legs,Too many fingers"

images = pipe(prompt=prompt, negative_prompt=negative_prompt, height=1024, width=1024).images[0]

✨ 主要特性

• 美學提升：能夠生成各種風格（如逼真攝影、插畫和圖形）的極具吸引力的圖像。
• 高提示對齊：確保生成的圖像能精確遵循用戶提供的文本描述，輸出更準確、有意義。
• 合法合規：由於使用了來自領先數據合作伙伴的100%授權數據進行訓練，該模型為版權和隱私侵權提供全面的法律責任保障，確保內容的道德使用。
• 歸因引擎：擁有專利的歸因引擎，可根據生成的圖像對數據合作伙伴進行公平補償。
• 可定製技術：提供源代碼和權重，方便進行廣泛的定製。

📦 安裝指南

使用Diffusers庫調用模型，需先安裝相關依賴：

pip install diffusers, hf_hub_download

💻 使用示例

基礎用法

pip install diffusers, hf_hub_download

from huggingface_hub import hf_hub_download
import os

try:
    local_dir = os.path.dirname(__file__)
except:
    local_dir = '.'
    
hf_hub_download(repo_id="briaai/BRIA-3.1", filename='pipeline_bria.py', local_dir=local_dir)
hf_hub_download(repo_id="briaai/BRIA-3.1", filename='transformer_bria.py', local_dir=local_dir)
hf_hub_download(repo_id="briaai/BRIA-3.1", filename='bria_utils.py', local_dir=local_dir)

import torch
from pipeline_bria import BriaPipeline

# trust_remote_code = True - allows loading a transformer which is not present at the transformers library(from transformer/bria_transformer.py)
pipe = BriaPipeline.from_pretrained("briaai/BRIA-3.1", torch_dtype=torch.bfloat16,trust_remote_code=True)
pipe.to(device="cuda")

prompt = "A portrait of a Beautiful and playful ethereal singer, golden designs, highly detailed, blurry background"
negative_prompt = "Logo,Watermark,Ugly,Morbid,Extra fingers,Poorly drawn hands,Mutation,Blurry,Extra limbs,Gross proportions,Missing arms,Mutated hands,Long neck,Duplicate,Mutilated,Mutilated hands,Poorly drawn face,Deformed,Bad anatomy,Cloned face,Malformed limbs,Missing legs,Too many fingers"

images = pipe(prompt=prompt, negative_prompt=negative_prompt, height=1024, width=1024).images[0]

高級用法

暫無高級用法示例。

📚 詳細文檔

獲取訪問權限

• API端點：Bria.ai
• ComfyUI：在工作流中使用
• 獲取BRIA 3.1權重：需要購買許可證才能訪問BRIA 3.1，以確保與數據合作伙伴的版稅管理和商業使用的全面責任覆蓋。
  • 如果你是初創企業或學生，可申請初創企業計劃來請求訪問。該計劃旨在用前沿技術支持新興企業和學術研究。
  • 立即聯繫我們，釋放BRIA 3.1的潛力！提交上述表單即表示你同意BRIA的隱私政策和條款與條件。
• 許可證：商業許可條款和條件

使用提示

1. 建議使用負提示。
2. 微調時，使用零值而非空文本嵌入。
3. 支持多種寬高比，但分辨率總體應約為1024*1024 = 100萬像素，例如：((1024,1024), (1280, 768), (1344, 768), (832, 1216), (1152, 832), (1216, 832), (960,1088)
4. 使用30 - 50步（步數越高越好）
5. 使用guidance_scale為5.0

🔧 技術細節

這些改進通過以下幾個關鍵技術升級實現： 首先，使用先進的視覺語言模型生成的合成字幕擴充了大型數據集。然後，通過集成最先進的變壓器（特別是使用MMDIT和DIT層）改進架構，並使用整流流目標進行訓練。這種方法與其他開放模型（如AuraFlow、Flux和SD3）類似。BRIA 3.1還採用2D RoPE進行位置嵌入、KQ歸一化以提高訓練穩定性，並使用噪聲偏移進行高分辨率訓練。

為確保推理和微調成本可控，BRIA 3.1設計得較為緊湊，由28個MMDIT層和8個DIT層組成，總共40億參數。僅使用T5文本編碼器，避免使用CLIP以減少不必要的偏差。在空間壓縮方面，使用開源的VAE f8，並確認該VAE不會給模型引入偏差。

基礎模型未進行蒸餾，原生支持無分類器引導，為微調提供了充分的靈活性。

此外，BRIA 3.1在多種寬高比和分辨率上進行訓練，能夠原生生成水平和垂直方向的100萬像素圖像。

最後，還為Diffusers代碼庫和ComfyUI提供全面支持，便於快速實驗和部署。微調代碼將很快提供。

📄 許可證

• 許可證類型：bria-t2i
• 許可證鏈接：商業許可條款和條件

⚠️ 重要提示

模型權重需購買商業許可證才能獲取。提交表單即表示同意BRIA的隱私政策和條款與條件。

💡 使用建議

1. 建議使用負提示。
2. 微調時，使用零值而非空文本嵌入。
3. 支持多種寬高比，但分辨率總體應約為1024*1024 = 100萬像素。
4. 使用30 - 50步（步數越高越好）。
5. 使用guidance_scale為5.0。