Slim205 commited on
Commit
6d95f34
·
verified ·
1 Parent(s): ff23930

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +23 -0
README.md CHANGED
@@ -26,4 +26,27 @@ model_inputs = [get_prompt(text)]
26
 
27
  print(model_inputs[0])
28
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
29
 
 
26
 
27
  print(model_inputs[0])
28
 
29
+ ### Example output :
30
+
31
+
32
+ <theorem>
33
+ theorem mathd_numbertheory_3 : (∑ x in Finset.range 10, (x + 1) ^ 2) % 10 = 5 := by
34
+ </theorem>
35
+ <conjecture>
36
+ theorem mathd_numbertheory_24 : ∑ k in Finset.range 11, k ^ 2 = 385 := by
37
+ </conjecture>
38
+
39
+ ### Inference
40
+ model = LLM(model=model_name, seed=1, trust_remote_code=True, swap_space=8, tensor_parallel_size=1, max_model_len=4096)
41
+
42
+ sampling_params = SamplingParams(
43
+ temperature=1,
44
+ max_tokens=2048,
45
+ top_p=0.95,
46
+ n=1,
47
+ )
48
+
49
+ model_outputs = model.generate(model_inputs, sampling_params, use_tqdm=True)
50
+ print(model_outputs[0].outputs[0].text)
51
+
52