fix img 9b6d513
Danilo Toapanta commited on
How to use danilotpnta/HuBERT-Genre-Clf with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("audio-classification", model="danilotpnta/HuBERT-Genre-Clf") # Load model directly
from transformers import AutoProcessor, AutoModelForAudioClassification
processor = AutoProcessor.from_pretrained("danilotpnta/HuBERT-Genre-Clf")
model = AutoModelForAudioClassification.from_pretrained("danilotpnta/HuBERT-Genre-Clf")How to use danilotpnta/HuBERT-Genre-Clf with Diffusers:
pip install -U diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("danilotpnta/HuBERT-Genre-Clf", dtype=torch.bfloat16, device_map="cuda")
prompt = "DistilHuBERT, Audio Encoder, Transfer Learning"
image = pipe(prompt).images[0]