Update README.md
Browse files
README.md
CHANGED
@@ -19,6 +19,7 @@ This project is for research purposes only. Third-party datasets may be subject
|
|
19 |
## How to Get Started with the Model
|
20 |
|
21 |
<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
|
|
|
22 |
import torch
|
23 |
|
24 |
from peft import AutoPeftModelForCausalLM
|
@@ -50,10 +51,10 @@ print(f"Context:\n{example[0]['content']}")
|
|
50 |
print(f"Original Answer:\n{example[2]['content']}")
|
51 |
|
52 |
print(f"Generated Answer:\n{outputs[0]['generated_text'][len(prompt):].strip()}")
|
53 |
-
|
54 |
|
55 |
## Training Details
|
56 |
-
|
57 |
Peft Config :
|
58 |
|
59 |
{
|
@@ -97,7 +98,7 @@ Hyperparameters:
|
|
97 |
"gradient_accumulation_steps" : 4
|
98 |
|
99 |
}
|
100 |
-
|
101 |
|
102 |
## Evaluation
|
103 |
|
|
|
19 |
## How to Get Started with the Model
|
20 |
|
21 |
<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
|
22 |
+
```
|
23 |
import torch
|
24 |
|
25 |
from peft import AutoPeftModelForCausalLM
|
|
|
51 |
print(f"Original Answer:\n{example[2]['content']}")
|
52 |
|
53 |
print(f"Generated Answer:\n{outputs[0]['generated_text'][len(prompt):].strip()}")
|
54 |
+
```
|
55 |
|
56 |
## Training Details
|
57 |
+
```
|
58 |
Peft Config :
|
59 |
|
60 |
{
|
|
|
98 |
"gradient_accumulation_steps" : 4
|
99 |
|
100 |
}
|
101 |
+
```
|
102 |
|
103 |
## Evaluation
|
104 |
|