from transformers import pipeline
question = "If you had a time machine, but could only go to the past or the future once and never return, which would you choose and why?"
generator = pipeline("text-generation", model="igorktech/Custom2", device="cuda")
output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
print(output["generated_text"])
Training procedure
This model was trained with SFT.
Framework versions
TRL: 0.21.0
Transformers: 4.55.0
Pytorch: 2.6.0
Datasets: 3.6.0
Tokenizers: 0.21.2
Citations
Cite TRL as:
@misc{vonwerra2022trl,
title = {{TRL: Transformer Reinforcement Learning}},
author = {Leandro von Werra and Younes Belkada and Lewis Tunstall and Edward Beeching and Tristan Thrush and Nathan Lambert and Shengyi Huang and Kashif Rasul and Quentin Gallou{\'e}dec},
year = 2020,
journal = {GitHub repository},
publisher = {GitHub},
howpublished = {\url{https://github.com/huggingface/trl}}
}
Runs of igorktech Custom2 on huggingface.co
20
Total runs
0
24-hour runs
0
3-day runs
0
7-day runs
0
30-day runs
More Information About Custom2 huggingface.co Model
Custom2 huggingface.co
Custom2 huggingface.co is an AI model on huggingface.co that provides Custom2's model effect (), which can be used instantly with this igorktech Custom2 model. huggingface.co supports a free trial of the Custom2 model, and also provides paid use of the Custom2. Support call Custom2 model through api, including Node.js, Python, http.
Custom2 huggingface.co is an online trial and call api platform, which integrates Custom2's modeling effects, including api services, and provides a free online trial of Custom2, you can try Custom2 online for free by clicking the link below.
igorktech Custom2 online free url in huggingface.co:
Custom2 is an open source model from GitHub that offers a free installation service, and any user can find Custom2 on GitHub to install. At the same time, huggingface.co provides the effect of Custom2 install, users can directly use Custom2 installed effect in huggingface.co for debugging and trial. It also supports api for free installation.