This is a repository where I have fine-tuned an Llama2 on an open source dataset from Kaggle.
The objective of this project is when you input an product name it will give you the department.
Product and description from Instacart Market Basket Analysis (https://www.kaggle.com/competitions/instacart-market-basket-analysis/data?select=products.csv.zip)
- Model used: 'TinyPixel/Llama-2-7B-bf16-sharded'