MEMO_onnx_runtime / README.md
cplusx's picture
Upload README.md with huggingface_hub
ef8d9aa verified
# MEMO ONNX Runtime Models
This repository stores exported ONNX Runtime model files for MEMO.
## Included Model Folders
- `memo_synthetic_early_fp16`: synthetic pretrained model, earlier checkpoint
- `memo_synthetic_late_fp16`: synthetic pretrained model, later checkpoint
- `memo_bsds_early_lora_fp16`: BSDS finetuned LoRA model
- `memo_bsds_late_lora_fp16`: BSDS finetuned LoRA model using the late synthetic base checkpoint
- `memo_biped_late_lora_fp16`: BIPED finetuned LoRA model using the late synthetic base checkpoint
Each folder contains:
- `dino_encoder_fp16.onnx`
- `memo_denoiser_fp16.onnx`