Diffusers Dataset Augmentation
Collection
Dataset augmentation project using diffuser SANA model • 3 items • Updated
How to use jgilewicz/SANA-LoRA-CIFAR10 with Diffusers:
pip install -U diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("Efficient-Large-Model/Sana_1600M_1024px_BF16_diffusers", dtype=torch.bfloat16, device_map="cuda")
pipe.load_lora_weights("jgilewicz/SANA-LoRA-CIFAR10")
prompt = "a low-resolution photo of an airplane, CIFAR-10 style"
image = pipe(prompt).images[0]How to use jgilewicz/SANA-LoRA-CIFAR10 with Sana:
# Load the model and infer image from text
import torch
from app.sana_pipeline import SanaPipeline
from torchvision.utils import save_image
sana = SanaPipeline("configs/sana_config/1024ms/Sana_1600M_img1024.yaml")
sana.from_pretrained("hf://jgilewicz/SANA-LoRA-CIFAR10")
image = sana(
prompt='a cyberpunk cat with a neon sign that says "Sana"',
height=1024,
width=1024,
guidance_scale=5.0,
pag_guidance_scale=2.0,
num_inference_steps=18,
) import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("Efficient-Large-Model/Sana_1600M_1024px_BF16_diffusers", dtype=torch.bfloat16, device_map="cuda")
pipe.load_lora_weights("jgilewicz/SANA-LoRA-CIFAR10")
prompt = "a low-resolution photo of an airplane, CIFAR-10 style"
image = pipe(prompt).images[0]



These are W1ndrunn3rr/SANA-LoRA-CIFAR10 DreamBooth LoRA weights for Efficient-Large-Model/Sana_1600M_1024px_BF16_diffusers.
The weights were trained using DreamBooth with the Sana diffusers trainer.
You should use a photo to trigger the image generation.
Download the *.safetensors LoRA in the Files & versions tab.
TODO
For more details, including weighting, merging and fusing LoRAs, check the documentation on loading LoRAs in diffusers
TODO
# TODO: add an example code snippet for running this diffusion pipeline
[TODO: provide examples of latent issues and potential remediations]
[TODO: describe the data used to train the model]
Unable to build the model tree, the base model loops to the model itself. Learn more.