---
tags:
- text-to-image
- lora
- diffusers
- template:diffusion-lora
- collage
- wallpaper
- vibrant
- multiple-panels
widget:
- text: >-
    A vibrant red spiderman is depicted in a collage of multiple panels. The
    background is a vibrant orange, adding a pop of color to the scene. The
    spidermans suit is a deep red, with a white stripe running down the center
    of his chest. His eyes are large, white, and his mouth is slightly open. His
    hands are positioned in a way that he is facing towards the right side of
    the frame. The entire frame is black, creating a striking contrast with the
    red and orange background.
  output:
    url: images/C1.webp
- text: >-
    A vibrant, black-suited Batman is depicted in a collage of multiple panels.
    The background is a deep, intense blue, adding a dramatic tone to the scene.
    Batmans suit is a rich black with a dark gray bat symbol on his chest. His
    eyes are narrow and white, and his mouth is set in a serious expression. His
    cape flows around him, and he is facing toward the right side of the frame,
    giving a sense of movement. The entire frame is outlined in dark gray,
    creating a bold contrast with the dark suit and blue background.
  output:
    url: images/C2.webp
- text: >-
    A vibrant, red-suited Deadpool is depicted in a collage of multiple panels.
    The background is a bold, electric yellow, adding an energetic vibe to the
    scene. Deadpool’s suit is a striking red with black patches around his eyes
    and across his torso, creating a dynamic look. His eyes are expressive and
    white, and his mouth is curved into a mischievous smirk. His hands are
    positioned confidently, holding one of his katanas, and he is facing toward
    the right side of the frame, exuding a playful, yet intense energy. The
    entire frame is outlined in black, creating a sharp contrast with the red
    suit and vibrant yellow background.
  output:
    url: images/L3.webp
base_model: black-forest-labs/FLUX.1-dev
instance_prompt: collage
license: creativeml-openrail-m
---
# Castor-Collage-Dim-Flux-LoRA

<Gallery />

**The model is still in the training phase. This is not the final version and may contain artifacts and perform poorly in some cases.**

## Model description 

**prithivMLmods/Castor-Collage-Dim-Flux-LoRA**

Image Processing Parameters 

| Parameter                 | Value  | Parameter                 | Value  |
|---------------------------|--------|---------------------------|--------|
| LR Scheduler              | constant | Noise Offset              | 0.03   |
| Optimizer                 | AdamW  | Multires Noise Discount   | 0.1    |
| Network Dim               | 64     | Multires Noise Iterations | 10     |
| Network Alpha             | 32     | Repeat & Steps           | 28 & 3.2k |
| Epoch                     | 12     | Save Every N Epochs       | 1      |

    Labeling: florence2-en(natural language & English)
    
    Total Images Used for Training : 27

## Setting Up
```
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-Collage-Dim-Flux-LoRA"
trigger_word = "collage"  # Leave trigger_word blank if not used.
pipe.load_lora_weights(lora_repo)

device = torch.device("cuda")
pipe.to(device)
```
## App File Structure

/project-root/

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

# Best Dimensions

- 512 X 512
- 1024 X 1024
- 768 X 1024

## Trigger words 🧨

> [!WARNING]
> **Trigger words:**  You should use `collage` to trigger the image generation.

## Download model

Weights for this model are available in Safetensors format.

[Download](/prithivMLmods/Castor-Collage-Dim-Flux-LoRA/tree/main) them in the Files & versions tab.