model_name: firered-lora
description: "FireRed-Edit-1.1 + Lightning LoRA — 3.2s/edit image editing"
python_version: py310

requirements:
  - torch==2.7.0
  - torchvision
  - diffusers>=0.33.0
  - transformers>=4.49.0
  - accelerate
  - safetensors
  - Pillow
  - peft
  - sentencepiece
  - protobuf

system_packages:
  - libgl1-mesa-glx
  - libglib2.0-0

resources:
  accelerator: H100
  use_gpu: true

runtime:
  predict_concurrency: 1

model_metadata:
  example_model_input:
    image: "<base64-encoded-image>"
    prompt: "Change the t-shirt to a formal black suit"
    steps: 6
    cfg: 1.0

secrets:
  hf_access_token: null
