Add pipeline tag, license, and remove redundant section

#1
by nielsr HF Staff - opened
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
+ ```