!pip install transformers datasets
from transformers import AutoModelForCausalLM, AutoTokenizer
Descarga el modelo desde Hugging Face
model_name = "EleutherAI/gpt-j-6B"
model = AutoModelForCausalLM.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)
Prueba generando texto
inputs = tokenizer("Escribe un texto sobre ", return_tensors="pt")
outputs = model.generate(inputs['input_ids'], max_length=50, num_return_sequences=1)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
