hieunguyen1053 commited on
Commit
241ed54
1 Parent(s): cdb8786

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -4
README.md CHANGED
@@ -16,12 +16,16 @@ model-index:
16
  name: Word prediction
17
  type: text-generation
18
  dataset:
19
- type: vlsp-2023-vllm/lambada
20
- name: ViLambada
21
  split: test
22
  metrics:
23
  - type: Perplexity
24
  value: 8.606673731963474
 
 
 
 
25
  ---
26
 
27
  # Hoa 7B (Bloom architecture)
@@ -49,5 +53,4 @@ input_ids = tokenizer(prompt, return_tensors="pt")['input_ids'].to(device)
49
  gen_tokens = model.generate(input_ids, max_length=max_length, repetition_penalty=1.1)
50
 
51
  print(tokenizer.batch_decode(gen_tokens)[0])
52
- ```
53
-
 
16
  name: Word prediction
17
  type: text-generation
18
  dataset:
19
+ type: vlsp-2023-vllm/vi_lambada
20
+ name: vi_lambada
21
  split: test
22
  metrics:
23
  - type: Perplexity
24
  value: 8.606673731963474
25
+ datasets:
26
+ - vlsp-2023-vllm/vi_lambada
27
+ metrics:
28
+ - perplexity
29
  ---
30
 
31
  # Hoa 7B (Bloom architecture)
 
53
  gen_tokens = model.generate(input_ids, max_length=max_length, repetition_penalty=1.1)
54
 
55
  print(tokenizer.batch_decode(gen_tokens)[0])
56
+ ```