Model Description
This model is used for text generation. if you want to use this model, try running the following code. Hope my model helps you.
Code
from transformers import AutoModelForCausalLM, AutoTokenizer
model_name = "FigoSans/TextGeneration"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name)
input_text = "Input Text" inputs = tokenizer(input_text, return_tensors="pt") outputs = model.generate(**inputs) generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
print(generated_text)