Update README.md
Browse files
README.md
CHANGED
|
@@ -5,7 +5,7 @@ datasets:
|
|
| 5 |
language:
|
| 6 |
- en
|
| 7 |
---
|
| 8 |
-
|
| 9 |
>python inference.py --temp 0.5 --top_k 64 --model_file lstm_H256.safetensors --start_sequence "User : what is the capital of France ? Bot :" --max_length 32
|
| 10 |
|
| 11 |
usually results in something like
|
|
|
|
| 5 |
language:
|
| 6 |
- en
|
| 7 |
---
|
| 8 |
+
the only files needed for inference is inference.py, word2idk.pkl, and lstm_Hxxx.safetensors
|
| 9 |
>python inference.py --temp 0.5 --top_k 64 --model_file lstm_H256.safetensors --start_sequence "User : what is the capital of France ? Bot :" --max_length 32
|
| 10 |
|
| 11 |
usually results in something like
|