Castor Red Dead Redemption 2 Flux LoRA

🚀 Castor-Red-Dead-Redemption-2-Flux-LoRA

本項目是為紀念《荒野大鏢客2》（Red Dead Redemption 2）發售六週年而推出的。它基於LoRA技術，用於生成與《荒野大鏢客2》相關的圖像。目前模型仍在訓練階段，並非最終版本，可能存在瑕疵，在某些情況下表現不佳。

🚀 快速開始

環境設置

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-Red-Dead-Redemption-2-Flux-LoRA"
trigger_word = "Red Dead Redemption 2"  # 若不使用觸發詞，可留空。
pipe.load_lora_weights(lora_repo)

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

✨ 主要特性

• 特定主題生成：能夠生成與《荒野大鏢客2》相關的圖像，如牛仔、西部場景等。
• 參數可調：提供了豐富的圖像生成參數，可根據需求進行調整。

📦 安裝指南

本項目的安裝主要是通過代碼導入相關庫和模型，具體步驟如下：

1. 確保你已經安裝了torch和pipelines庫。
2. 按照上述“環境設置”部分的代碼，導入基礎模型和LoRA模型。

💻 使用示例

基礎用法

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-Red-Dead-Redemption-2-Flux-LoRA"
trigger_word = "Red Dead Redemption 2"  # 若不使用觸發詞，可留空。
pipe.load_lora_weights(lora_repo)

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

# 生成圖像的代碼示例
prompt = "Red Dead Redemption 2, a cowboy on a horse"
image = pipe(prompt).images[0]
image.save("generated_image.png")

高級用法

在高級用法中，你可以調整更多的參數來生成不同風格和質量的圖像。例如：

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-Red-Dead-Redemption-2-Flux-LoRA"
trigger_word = "Red Dead Redemption 2"  # 若不使用觸發詞，可留空。
pipe.load_lora_weights(lora_repo)

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

# 調整參數的高級用法示例
prompt = "Red Dead Redemption 2, a cowboy on a horse, sunset"
num_inference_steps = 50  # 推理步數
guidance_scale = 7.5  # 引導比例
image = pipe(prompt, num_inference_steps=num_inference_steps, guidance_scale=guidance_scale).images[0]
image.save("generated_image_advanced.png")

📚 詳細文檔

模型參數詳情

標註信息

標註使用florence2 - en（自然語言 & 英語）。

訓練數據

總共使用了19張圖像進行訓練。

應用文件結構

/project-root/
├── .gitattributes         
├── README.md      
├── app.py             
├── pythonproject.py     

最佳尺寸

• 512 X 512
• 1024 X 1024
• 768 X 1024

觸發詞

⚠️ 重要提示

你應該使用Red Dead Redemption 2來觸發圖像生成。

模型下載

本模型的權重以Safetensors格式提供。 點擊下載，可在“Files & versions”標籤中找到。

SDXL LoRA

相關鏈接：https://huggingface.co/prithivMLmods/Canopus-Red-Dead-Minimalize-LoRA

📄 許可證

本項目採用Apache 2.0許可證。