from transformers import pipeline #pipe = pipeline("text-generation", model="TheBloke/llava-v1.5-13B-AWQ",device_map="cuda:1") pipe = pipeline("image_classification", model="TheBloke/llava-v1.5-13B-AWQ",device_map="cuda:1") images="https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/pipeline-cat-chonk.jpeg" pipe(images)