Skip to content

Instantly share code, notes, and snippets.

@irr
irr / flux_lora_cfg.py
Created January 4, 2025 15:03 — forked from AmericanPresidentJimmyCarter/flux_lora_cfg.py
Use your flux-dev LoRA with a quantized model and CFG in <16gb VRAM
import inspect
from typing import Any, Callable, Dict, List, Optional, Union
import numpy as np
import torch
from transformers import CLIPTextModel, CLIPTokenizer, T5EncoderModel, T5TokenizerFast
from diffusers.image_processor import VaeImageProcessor
from diffusers.loaders import FluxLoraLoaderMixin
from diffusers.models.autoencoders import AutoencoderKL