learn-abc commited on
Commit
2bee241
·
verified ·
1 Parent(s): 91ae1a2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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" # Replace with your actual repo ID
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