Image Classification
Transformers
PyTorch
Safetensors
vit
anime
quality assurance
dataset maintenance
Instructions to use shadowlilac/aesthetic-shadow with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use shadowlilac/aesthetic-shadow with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="shadowlilac/aesthetic-shadow") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")# Load model directly from transformers import AutoImageProcessor, AutoModelForImageClassification processor = AutoImageProcessor.from_pretrained("shadowlilac/aesthetic-shadow") model = AutoModelForImageClassification.from_pretrained("shadowlilac/aesthetic-shadow") - Notebooks
- Google Colab
- Kaggle
run without transformers pipeline
#2
by csegalin - opened
is possible to run the model without the transformer pipeline?
asking as it doesn't allow to run on multi gpus for mutliprocessing and using cpu is quite slow
You can run on GPU by moving the model to GPU. one way would be to have a simple subprocess launcher so that it runs on your cluster/setup