Update README.md
Browse files
README.md
CHANGED
|
@@ -19,7 +19,7 @@ This repository contains a fine-tuned version of the `unsloth/tinyllama-chat-bnb
|
|
| 19 |
|
| 20 |
- **Base Model:** `unsloth/tinyllama-chat-bnb-4bit`
|
| 21 |
- **Fine-tuning Method:** LoRA (Low-Rank Adaptation)
|
| 22 |
-
- **Trained on:** A custom dataset of HTML product snippets and their corresponding JSON representations.
|
| 23 |
|
| 24 |
## Usage
|
| 25 |
|
|
@@ -33,7 +33,7 @@ from unsloth import FastLanguageModel
|
|
| 33 |
import torch
|
| 34 |
import json
|
| 35 |
|
| 36 |
-
model_name = "learn-abc/html-model-tinyllama-chat-bnb-4bit" #
|
| 37 |
max_seq_length = 2048 # Or your chosen sequence length
|
| 38 |
dtype = None # Auto detection
|
| 39 |
|
|
|
|
| 19 |
|
| 20 |
- **Base Model:** `unsloth/tinyllama-chat-bnb-4bit`
|
| 21 |
- **Fine-tuning Method:** LoRA (Low-Rank Adaptation)
|
| 22 |
+
- **Trained on:** A custom dataset `json_extraction_dataset_500.json` of HTML product snippets and their corresponding JSON representations.
|
| 23 |
|
| 24 |
## Usage
|
| 25 |
|
|
|
|
| 33 |
import torch
|
| 34 |
import json
|
| 35 |
|
| 36 |
+
model_name = "learn-abc/html-model-tinyllama-chat-bnb-4bit" # Hugging face model repo ID
|
| 37 |
max_seq_length = 2048 # Or your chosen sequence length
|
| 38 |
dtype = None # Auto detection
|
| 39 |
|