Tentative update to use transformers pipeline
#1
by KoichiYasuoka - opened
To use transformers pipeline with trust_remote_code=True:
from transformers import pipeline
nlp = pipeline("text-generation", "JonusNattapong/AiDaeng-Thai-RoPE", trust_remote_code=True)
text = "สวัสดีครับ"
response = nlp(text, max_new_tokens=50, temperature=0.7)[0]["generated_text"]
print(response)
JonusNattapong changed pull request status to merged