Image-to-Image
Diffusers
Safetensors
super-resolution
image-super-resolution
flux
lora
dpo
diffusion
Instructions to use ngoctham/ASASR with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use ngoctham/ASASR with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline from diffusers.utils import load_image # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("fill-in-base-model", torch_dtype=torch.bfloat16, device_map="cuda") pipe.load_lora_weights("ngoctham/ASASR") prompt = "Turn this cat into a dog" input_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png") image = pipe(image=input_image, prompt=prompt).images[0] - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- Draw Things
File size: 1,956 Bytes
293c385 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 | ---
license: cc-by-nc-4.0
tags:
- super-resolution
- image-super-resolution
- flux
- lora
- dpo
- diffusion
library_name: diffusers
pipeline_tag: image-to-image
---
# ASASR — Coloring the Noise: Adversarial Sobolev Alignment for Faithful Image Super-Resolution
Pretrained weights for the ICML 2026 paper
**[Coloring the Noise: Adversarial Sobolev Alignment for Faithful Image Super-Resolution](https://arxiv.org/abs/2605.23264)**
(Hongbo Wang, Huaibo Huang, Pin Wang, Jinhua Hao, Chao Zhou, Ran He).
➡️ **Code & full instructions: https://github.com/wafer-bob/ASASR**
ASASR performs ×4 image super-resolution with a **FLUX.1-dev** backbone and dual-LoRA
inference: a base **SR LoRA** (upscaling prior, OminiControl) plus our **DPO LoRA** trained
with a Sobolev frequency-weighted, adversarially-guided DPO objective (AS-DPO).
## Files
| File | Size | Use |
|---|---|---|
| `sr_lora/pytorch_lora_weights_v2.safetensors` | ~885 MB | base SR LoRA — **inference** |
| `dpo_lora/adapter_model.safetensors` | ~111 MB | ASASR AS-DPO LoRA — **inference** |
| `adv_lora/adapter_model.safetensors` | ~111 MB | rank-16 AMG adversary — **training only** |
## Download
```bash
huggingface-cli download wafer-bob/ASASR --local-dir ./checkpoints
```
Then follow the [GitHub README](https://github.com/wafer-bob/ASASR) for inference
(`bash scripts/infer.sh`) and training.
## License
This project is released under [CC-BY-NC-4.0](https://creativecommons.org/licenses/by-nc/4.0/) for **non-commercial research use only**.
Copyright (c) 2026 The Authors and Kuaishou Technology.
## Citation
```bibtex
@inproceedings{wang2026asasr,
title = {Coloring the Noise: Adversarial Sobolev Alignment for Faithful Image Super-Resolution},
author = {Wang, Hongbo and Huang, Huaibo and Wang, Pin and Hao, Jinhua and Zhou, Chao and He, Ran},
booktitle = {International Conference on Machine Learning (ICML)},
year = {2026}
}
```
|