Aller au contenu

agent Banner

Installation

Prérequis

  • Python 3.8+
  • PyTorch 2.0+
  • Transformers 4.25+
  • accelerate (optionnel)

Installation des dépendances

pip install torch transformers accelerate

Télécharger agentV1

from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("gopu-poss/agentV1")
model = AutoModelForCausalLM.from_pretrained(
    "gopu-poss/agentV1",
    device_map="auto",
    torch_dtype="auto"
)
Vérifier linstallation
print(model.config)

verification*

```python print(model.config)