tinygrad Showcase — 비전·언어·생성 모델 실행

작은 라이브러리임에도 tinygrad는 정말 많은 것을 할 수 있어요. 최신 비전 모델부터 언어 모델까지, examples 디렉터리의 스크립트로 실제 최신 모델을 돌려볼 수 있습니다.

출처: https://docs.tinygrad.org/showcase

Vision

EfficientNet

이미지 URL을 넘겨 그게 무엇인지 알아내거나, 웹캠으로 앞에 뭐가 있는지 감지할 수 있어요.

python3 examples/efficientnet.py ./test/models/efficientnet/Chicken.jpg
# or, if you have a camera and OpenCV installed
python3 examples/efficientnet.py webcam

YOLOv8

examples/yolov8.py 를 살펴보세요.

Audio

Whisper

examples/whisper.py 를 사용해요. pyaudio와 torchaudio가 설치돼 있어야 합니다.

SMALL=1 python3 examples/whisper.py

Generative

Stable Diffusion

python3 examples/stable_diffusion.py

LLaMA

LLaMA 가중치를 weights/LLaMA 디렉터리로 내려받아 넣어야 합니다(디렉터리는 만들어야 할 수 있어요). 그러면 Stacy와 대화할 수 있어요.

python3 examples/llama.py

Conversation

espeak 설치와 PHONEMIZER_ESPEAK_LIBRARY 설정이 필요해요.

python3 examples/conversation.py

더 알아보기

퀵스타트는 Quickstart 참고
MNIST 튜토리얼은 MNIST Tutorial 참고
예시 코드는 GitHub tinygrad/tinygrad/examples 참고