pbevan11
/

llama-3.1-8b-ocr-correction-merged

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

llama-3.1-8b-ocr-correction-merged / cog.yaml

pbevan11's picture

Upload 2 files

383292f verified 3 months ago

history blame contribute delete

No virus

240 Bytes

	build:
	gpu: true
	python_version: "3.10"
	system_packages:
	- "libgl1-mesa-glx"
	- "libglib2.0-0"
	python_packages:
	- "torch==2.0.1"
	- "transformers==4.30.2"
	- "peft==0.4.0"

	predict: "predict.py:Predictor"