How to use aldigobbler/ben2_30m_test_distilled_e1 with BEN2:
import requests from PIL import Image from ben2 import AutoModel url = "https://huggingface.co/datasets/mishig/sample_images/resolve/main/teapot.jpg" image = Image.open(requests.get(url, stream=True).raw) model = AutoModel.from_pretrained("aldigobbler/ben2_30m_test_distilled_e1") model.to("cuda").eval() foreground = model.inference(image)
Original parameters: 94,134,658
Pruned parameters: 36,275,254
Parameter reduction: 61.5%
distill loss: 0.0008
Files info
import requests from PIL import Image from ben2 import AutoModel url = "https://huggingface.co/datasets/mishig/sample_images/resolve/main/teapot.jpg" image = Image.open(requests.get(url, stream=True).raw) model = AutoModel.from_pretrained("aldigobbler/ben2_30m_test_distilled_e1") model.to("cuda").eval() foreground = model.inference(image)