Add pipeline tag, license, and remove redundant section
#1
by
nielsr
HF Staff
- opened
README.md
CHANGED
|
@@ -1,7 +1,9 @@
|
|
| 1 |
---
|
| 2 |
library_name: transformers
|
| 3 |
tags:
|
| 4 |
-
|
|
|
|
|
|
|
| 5 |
---
|
| 6 |
|
| 7 |
# LagKV Cache
|
|
@@ -48,4 +50,5 @@ gen_out = model.generate(
|
|
| 48 |
trust_remote_code=True,
|
| 49 |
)
|
| 50 |
print(tokenizer.batch_decode(gen_out.sequences, skip_special_tokens=True))
|
| 51 |
-
assert "lagkvcache" in str(type(gen_out.past_key_values)).lower()
|
|
|
|
|
|
| 1 |
---
|
| 2 |
library_name: transformers
|
| 3 |
tags:
|
| 4 |
+
- custom_generate
|
| 5 |
+
license: mit
|
| 6 |
+
pipeline_tag: text-generation
|
| 7 |
---
|
| 8 |
|
| 9 |
# LagKV Cache
|
|
|
|
| 50 |
trust_remote_code=True,
|
| 51 |
)
|
| 52 |
print(tokenizer.batch_decode(gen_out.sequences, skip_special_tokens=True))
|
| 53 |
+
assert "lagkvcache" in str(type(gen_out.past_key_values)).lower()
|
| 54 |
+
```
|