12 lines
354 B
Python
12 lines
354 B
Python
|
|
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)
|
|
|
|
|
|
|