Skip to content

Releases: gauthamnair2005/donutllmcore

Donut LLM Core Library 1.0.0

28 May 13:49
dde0d4e
Compare
Choose a tag to compare

Donut LLM Core Library

This library will be used by future versions of Donut-LLM-Tools, instead of CreateLLM library.

Usage

LLM Creation

import DonutLLMCore
path = input("Enter path to dataset : ")
model = DonutLLMCore.ModelTrainer(path,max_iters=100)
model = model.trainer()

LLM Usage

import DonutLLMCore
path = input("Enter the path of the model : ")
model = DonutLLMCore.LLMModel(path)
prompt = input("Enter prompt : ")
print(model.generate(prompt))