yuuhan

/

videomae-base-lora-ucf101-subset

Like 0

# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("video-classification", model="yuuhan/videomae-base-lora-ucf101-subset")

# pip install -U transformers accelerate
# Load model directly
from transformers import AutoImageProcessor, AutoModelForVideoClassification

processor = AutoImageProcessor.from_pretrained("yuuhan/videomae-base-lora-ucf101-subset")
model = AutoModelForVideoClassification.from_pretrained("yuuhan/videomae-base-lora-ucf101-subset", device_map="auto")