Qndhm commited on
Commit
fc32c34
·
verified ·
1 Parent(s): eee15cb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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 involved since the layer keys must be aligned correctly)*:
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