类库
› TinySR
Microtreei/TinySR
TinySR是CVPR 2026论文“TinySR: Pruning Diffusion for Real-World Image Super-Resolution”的官方实现。该仓库基于PyTorch,利用DiT架构的剪枝方案,旨在解决现实世界图像超分辨率问题,提供推理代码和预训练模型。
技术栈
根目录 python
查看全部依赖 (11)
依赖
NumPy
accelerate
diffusers
loralib
peft
pillow
pyiqa
torch
torchvision
transformers
wandb