mhenrichsen commited on
Commit
67d8cc7
·
1 Parent(s): 2f658dd

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +33 -3
README.md CHANGED
@@ -26,11 +26,41 @@ Modellen er trænet med ChatML format (samme som OpenAI's modeller), og skal bru
26
 
27
 
28
 
29
- More information needed
30
 
31
- ## Training and evaluation data
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
32
 
33
- More information needed
34
 
35
  ## Training procedure
36
 
 
26
 
27
 
28
 
29
+ ## Inferens
30
 
31
+ vLLM
32
+ ```
33
+ pip install vllm
34
+ ```
35
+
36
+
37
+ ```python
38
+ from vllm import LLM, SamplingParams
39
+
40
+ sampling_params = SamplingParams(temperature=0.8, top_p=0.95, max_tokens=1024)
41
+ llm = LLM(model="mhenrichsen/danskgpt-tiny-chat")
42
+
43
+ system_message = "Du er en hjælpsom assistent."
44
+ conversation_history = f"<|im_start|>system\n{system_message}<|im_end|>\n<|im_start|>user\n"
45
+ while True:
46
+ prompt = input("Bruger: ")
47
+ new_prompt = f"{conversation_history}{prompt}<|im_end|>\n<|im_start|>assistant\n"
48
+
49
+ outputs = llm.generate(new_prompt, sampling_params)
50
+ for output in outputs:
51
+ prompt = output.prompt
52
+ generated_text = output.outputs[0].text
53
+ print(f"AI: {generated_text!r}")
54
+ conversation_history = f"{prompt}{generated_text!r}<|im_end|>\n<|im_start|>user\n"
55
+
56
+ ```
57
+
58
+ eller som endpoint:
59
+
60
+ ```
61
+ python -m vllm.entrypoints.openai.api_server --model mhenrichsen/danskgpt-tiny-chat
62
+ ```
63
 
 
64
 
65
  ## Training procedure
66