Skip to content

README.md

Latest commit

 

History

History
35 lines (24 loc) · 735 Bytes

README.md

File metadata and controls

35 lines (24 loc) · 735 Bytes
Raw

Examples

Example scripts for the nonescape Python library.

Scripts

basic_inference.py

Basic image classification example.

python basic_inference.py model.safetensors image.jpg

aria_test.py

ARIA dataset evaluation script with multi-GPU support.

# Single GPU
python aria_test.py model.safetensors --data-path ./aria_dataset

# Multi-GPU
torchrun --nproc_per_node=gpu aria_test.py model.safetensors --data-path ./aria_dataset

Model Setup

Download models before running examples:

# Full model
wget https://nonescape.sfo2.cdn.digitaloceanspaces.com/nonescape-v0.safetensors

# Mini model  
wget https://nonescape.sfo2.cdn.digitaloceanspaces.com/nonescape-mini-v0.safetensors