nielsr HF Staff commited on
Commit
fdbfb59
·
verified ·
1 Parent(s): dafad47

Add pipeline tag, license, and remove redundant section

Browse files

This PR improves the model card by:
- Adding `pipeline_tag: text-generation` to ensure better discoverability for users looking for text generation models or related optimizations.
- Adding `license: mit` to the metadata.
- Removing the extraneous `# File information` section from the content, as it's not part of a standard model card.

Files changed (1) hide show
  1. README.md +5 -2
README.md CHANGED
@@ -1,7 +1,9 @@
1
  ---
2
  library_name: transformers
3
  tags:
4
- - custom_generate
 
 
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
+ ```