Chat-With-YourFAV-Character-Small
This model is a conversational agent fine-tuned on the game script of "The World Ends With You" for generating dialogues in the style of the character Joshua. It is based on the DialoGPT-small architecture and further trained to mimic the conversational patterns and style of Joshua, offering users the ability to interact with a simulated version of the character.
Model Details
Developed by
- [Manan Raval- Virtual Height IT Services Pvt. Ltd.]
Model Description
This model was developed to explore the capabilities of language models in generating character-specific dialogues. It utilizes the transformer architecture for generating text that is coherent and contextually relevant to the inputs it receives.
-
Model type: DialoGPT-small
-
Language(s) (NLP): English
-
License: Apache-2.0
-
Sources
-
Finetuned from: microsoft/DialoGPT-small
Usage
Direct Use
This model is intended for direct interaction by users who want to engage in conversation with an AI that simulates the character Joshua from "The World Ends With You." It can be integrated into chat applications, games, or any platform where conversational AI can enhance user experience.
Out-of-Scope Use
This model is not designed for high-stakes decision-making processes, sensitive contexts, or as a replacement for human interaction in therapeutic or counseling settings.
Bias, Risks, and Limitations
The dialogues generated by this model are based on the character's portrayal in a specific video game and might not be appropriate for all contexts. Users should be aware of the potential for generating unexpected or offensive content.
Recommendations
It is recommended to monitor the interactions with the model and implement appropriate filtering mechanisms to prevent misuse or the generation of harmful content.
How to Get Started
from transformers import AutoModelForCausalLM, AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained("mananvh/Chat-With-YourFAV-Character-Small")
model = AutoModelForCausalLM.from_pretrained("mananvh/Chat-With-YourFAV-Character-Small")
# For interactive conversation
input_text = "Your input text here"
input_ids = tokenizer.encode(input_text + tokenizer.eos_token, return_tensors='pt')
chat_history_ids = model.generate(input_ids, max_length=1000, pad_token_id=tokenizer.eos_token_id)
response = tokenizer.decode(chat_history_ids[:, input_ids.shape[-1]:][0], skip_special_tokens=True)
print(response)
Training and Evaluation
Training Data
The model was fine-tuned on a dataset extracted from the game script of "The World Ends With You," focusing specifically on dialogues involving the character Joshua.
Training Procedure
The model was trained using the Hugging Face Transformers library with the following hyperparameters: learning rate of 5e-5, batch size of 4, and a total of 4 epochs. The training was performed on a single GPU.
Evaluation
Due to the conversational nature of the model, its performance was primarily evaluated based on qualitative assessments of its ability to generate coherent and character-consistent dialogues.
Technical Specifications
Model Architecture
The model is based on the DialoGPT-small architecture, which is a variant of the GPT-2 model optimized for dialogue generation tasks.
Compute Infrastructure
The training was conducted on a single GPU setup, utilizing NVIDIA CUDA for acceleration. YourModelNameHere
Citation
@misc{chatprompt_ai_2024,
title={ChatPrompt-AI: A Conversational Agent Simulating Joshua from TWEWY},
author={Manan},
year={2024},
howpublished={\url{https://huggingface.co/mananvh/Chat-With-YourFAV-Character-Small}}
}
Contact
For any questions or feedback regarding this model, please contact [mananraval.vhits@gmail.com].