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许可证。