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