Update README.md
Browse files
README.md
CHANGED
@@ -37,8 +37,8 @@ pipeline_tag: text-generation
|
|
37 |
{Assistant}
|
38 |
```
|
39 |
### Usage
|
40 |
-
|
41 |
-
|
42 |
|
43 |
```python
|
44 |
import torch
|
@@ -59,7 +59,6 @@ output = model.generate(**inputs, streamer=streamer, use_cache=True, max_new_tok
|
|
59 |
output_text = tokenizer.decode(output[0], skip_special_tokens=True)
|
60 |
```
|
61 |
|
62 |
-
Our model can handle >10k input tokens thanks to the `rope_scaling` option.
|
63 |
|
64 |
## Hardware and Software
|
65 |
|
|
|
37 |
{Assistant}
|
38 |
```
|
39 |
### Usage
|
40 |
+
- Tested on A100 80GB
|
41 |
+
- Our model can handle >10k input tokens thanks to the `rope_scaling` option
|
42 |
|
43 |
```python
|
44 |
import torch
|
|
|
59 |
output_text = tokenizer.decode(output[0], skip_special_tokens=True)
|
60 |
```
|
61 |
|
|
|
62 |
|
63 |
## Hardware and Software
|
64 |
|