Update README.md
Browse files
README.md
CHANGED
|
@@ -1,5 +1,8 @@
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
|
|
|
|
|
|
|
|
|
| 3 |
base_model: ibm-granite/granite-3.2-8b-instruct
|
| 4 |
library_name: peft
|
| 5 |
---
|
|
@@ -48,7 +51,7 @@ from peft import PeftModel, PeftConfig
|
|
| 48 |
import json
|
| 49 |
|
| 50 |
BASE_NAME = "ibm-granite/granite-3.2-8b-instruct"
|
| 51 |
-
LORA_NAME = "ibm-granite/granite-
|
| 52 |
device=torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
| 53 |
|
| 54 |
tokenizer = AutoTokenizer.from_pretrained(BASE_NAME, padding_side='left', trust_remote_code=True)
|
|
@@ -175,4 +178,4 @@ Notes:
|
|
| 175 |
## Model Card Authors
|
| 176 |
|
| 177 |
[Yannis Katsis](mailto:[email protected])</br>
|
| 178 |
-
[Chulaka Gunasekara](mailto:[email protected])
|
|
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
| 3 |
+
language:
|
| 4 |
+
- en
|
| 5 |
+
pipeline_tag: text-generation
|
| 6 |
base_model: ibm-granite/granite-3.2-8b-instruct
|
| 7 |
library_name: peft
|
| 8 |
---
|
|
|
|
| 51 |
import json
|
| 52 |
|
| 53 |
BASE_NAME = "ibm-granite/granite-3.2-8b-instruct"
|
| 54 |
+
LORA_NAME = "ibm-granite/granite-3.2-8b-lora-rag-citation-generation"
|
| 55 |
device=torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
| 56 |
|
| 57 |
tokenizer = AutoTokenizer.from_pretrained(BASE_NAME, padding_side='left', trust_remote_code=True)
|
|
|
|
| 178 |
## Model Card Authors
|
| 179 |
|
| 180 |
[Yannis Katsis](mailto:[email protected])</br>
|
| 181 |
+
[Chulaka Gunasekara](mailto:[email protected])
|