DSULT-Core/ShareGPT-X
Viewer • Updated • 109k • 319 • 17
How to use Mauricio-100/agent-ai with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("token-classification", model="Mauricio-100/agent-ai") # Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("Mauricio-100/agent-ai")
model = AutoModelForCausalLM.from_pretrained("Mauricio-100/agent-ai")Agent-AI est un modèle de génération de texte basé sur GPT-NeoX, conçu pour être léger, modulaire et compatible avec Xenova et Hugging Face Transformers. Il peut être utilisé pour créer des assistants conversationnels, des agents intelligents et tout type d’application de génération de texte.
Clone le dépôt et installe les dépendances Python :
git clone https://github.com/Mauricio-100/agent-ai.git
cd agent-ai
pip install -r requirements.txt
⚠️ Compatible Python 3.10+
from transformers import pipeline
# Charger le modèle depuis le dossier local ou HF hub
bot = pipeline("text-generation", model="./agent_model")
# Générer du texte
response = bot("Bonjour, comment vas-tu ?", max_length=100)
print(response[0]['generated_text'])
agent-ai/
│
├─ agent_model/ # Modèle pré-entraîné
│ ├─ pytorch_model.bin
│ └─ config.json
│
├─ agent_tokenizer/ # Tokenizer pour convertir texte ↔ tokens
│ ├─ tokenizer.json
│ └─ vocab.json # avenir ->
│
└─ README.md # Ce fichier
git checkout -b feature/my-featuregit commit -am 'Add new feature'git push origin feature/my-featureBase model
trl-internal-testing/tiny-GPTNeoXForCausalLM