Transformers How to use Humaid-alblooshi/bert-train-6layer-optimized-5-epoch with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="Humaid-alblooshi/bert-train-6layer-optimized-5-epoch") # pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("Humaid-alblooshi/bert-train-6layer-optimized-5-epoch")
model = AutoModelForSequenceClassification.from_pretrained("Humaid-alblooshi/bert-train-6layer-optimized-5-epoch", device_map="auto") 
