How to use from the
Use from the
Diffusers library
pip install -U diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline

# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("HiDream-ai/HiDream-I1-Dev", dtype=torch.bfloat16, device_map="cuda")
pipe.load_lora_weights("Nomnoos/trained-hidream-lora")

prompt = "a 3dicon, a llama eating ramen"
image = pipe(prompt).images[0]

HiDream Image DreamBooth LoRA - Nomnoos/trained-hidream-lora

Prompt
a 3dicon, a llama eating ramen
Prompt
a 3dicon, a llama eating ramen
Prompt
a 3dicon, a llama eating ramen
Prompt
a 3dicon, a llama eating ramen

Model description

These are Nomnoos/trained-hidream-lora DreamBooth LoRA weights for HiDream-ai/HiDream-I1-Dev.

The weights were trained using DreamBooth with the HiDream Image diffusers trainer.

Trigger words

You should use 3d icon to trigger the image generation.

Download model

Download the *.safetensors LoRA in the Files & versions tab.

Use it with the 🧨 diffusers library

    >>> import torch
    >>> from transformers import PreTrainedTokenizerFast, LlamaForCausalLM
    >>> from diffusers import HiDreamImagePipeline

    >>> tokenizer_4 = PreTrainedTokenizerFast.from_pretrained("meta-llama/Meta-Llama-3.1-8B-Instruct")
    >>> text_encoder_4 = LlamaForCausalLM.from_pretrained(
    ...     "meta-llama/Meta-Llama-3.1-8B-Instruct",
    ...     output_hidden_states=True,
    ...     output_attentions=True,
    ...     torch_dtype=torch.bfloat16,
    ... )

    >>> pipe = HiDreamImagePipeline.from_pretrained(
    ...     "HiDream-ai/HiDream-I1-Full",
    ...     tokenizer_4=tokenizer_4,
    ...     text_encoder_4=text_encoder_4,
    ...     torch_dtype=torch.bfloat16,
    ... )
    >>> pipe.enable_model_cpu_offload()
    >>> pipe.load_lora_weights(f"Nomnoos/trained-hidream-lora")
    >>> image = pipe(f"3d icon").images[0]

For more details, including weighting, merging and fusing LoRAs, check the documentation on loading LoRAs in diffusers

Intended uses & limitations

How to use

# TODO: add an example code snippet for running this diffusion pipeline

Limitations and bias

[TODO: provide examples of latent issues and potential remediations]

Training details

[TODO: describe the data used to train the model]

Downloads last month
12
Inference Examples
Examples
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for Nomnoos/trained-hidream-lora

Adapter
(3)
this model