google/quickdraw
Updated • 409 • 29
How to use WinKawaks/SketchXAI-Base-QuickDraw30 with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoImageProcessor, ViTForSketchClassification
processor = AutoImageProcessor.from_pretrained("WinKawaks/SketchXAI-Base-QuickDraw30")
model = ViTForSketchClassification.from_pretrained("WinKawaks/SketchXAI-Base-QuickDraw30", device_map="auto")A full description of this project can be found at https://sketchxai.github.io/.
# pip install -U transformers accelerate # Load model directly from transformers import AutoImageProcessor, ViTForSketchClassification processor = AutoImageProcessor.from_pretrained("WinKawaks/SketchXAI-Base-QuickDraw30") model = ViTForSketchClassification.from_pretrained("WinKawaks/SketchXAI-Base-QuickDraw30", device_map="auto")