Skip to content

Donut LLM Core Library 1.0.0

Latest
Compare
Choose a tag to compare
@gauthamnair2005 gauthamnair2005 released this 28 May 13:49
dde0d4e

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))