席亚东
commited on
Commit
·
7a8fd9d
1
Parent(s):
16bf127
update the readme
Browse files
README.md
CHANGED
|
@@ -75,7 +75,7 @@ After the environment is ready, several lines of codes can realize the inference
|
|
| 75 |
|
| 76 |
``` python
|
| 77 |
|
| 78 |
-
from inference import Inference
|
| 79 |
model_path = "transformer_lm_gpt2_xxl_dialogue/checkpoint_best.pt"
|
| 80 |
data_path = "transformer_lm_gpt2_xxl_dialogue"
|
| 81 |
eet_batch_size = 10 # max inference batch size, adjust according to cuda memory, 40GB memory is necessary
|
|
|
|
| 75 |
|
| 76 |
``` python
|
| 77 |
|
| 78 |
+
from inference import Inference, Dialogue
|
| 79 |
model_path = "transformer_lm_gpt2_xxl_dialogue/checkpoint_best.pt"
|
| 80 |
data_path = "transformer_lm_gpt2_xxl_dialogue"
|
| 81 |
eet_batch_size = 10 # max inference batch size, adjust according to cuda memory, 40GB memory is necessary
|