Initial commit 7ff52a6
范昊天 commited on
How to use DY-Evalab/one-align_ep2_bs64_formatPrompt_hn_v2 with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("feature-extraction", model="DY-Evalab/one-align_ep2_bs64_formatPrompt_hn_v2", trust_remote_code=True) # Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("DY-Evalab/one-align_ep2_bs64_formatPrompt_hn_v2", trust_remote_code=True, device_map="auto")