Update README.md
Browse files
README.md
CHANGED
|
@@ -13,8 +13,8 @@ pipeline_tag: text-classification
|
|
| 13 |
- Fine-tuning was performed efficiently using **LoRA (Low-Rank Adaptation)** from the PEFT library, with training managed by the **PyTorch Lightning** framework.
|
| 14 |
- The **LoRA adapter weights** must be combined with the base DistilBERT model weights for inference.
|
| 15 |
|
| 16 |
-
## You can use the following Python code to perform inference
|
| 17 |
-
*(The code is
|
| 18 |
|
| 19 |
```python
|
| 20 |
import torch
|
|
|
|
| 13 |
- Fine-tuning was performed efficiently using **LoRA (Low-Rank Adaptation)** from the PEFT library, with training managed by the **PyTorch Lightning** framework.
|
| 14 |
- The **LoRA adapter weights** must be combined with the base DistilBERT model weights for inference.
|
| 15 |
|
| 16 |
+
## You can use the following Python code to perform inference:
|
| 17 |
+
*(The code is lengthy since the layer keys must be aligned correctly)*
|
| 18 |
|
| 19 |
```python
|
| 20 |
import torch
|