neighborwang commited on
Commit
bfef9e7
1 Parent(s): 675e7a5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -31
README.md CHANGED
@@ -9,34 +9,4 @@ language:
9
  pipeline_tag: text2text-generation
10
  ---
11
 
12
- # Model Trained Using AutoTrain
13
-
14
- This model was trained using AutoTrain. For more information, please visit [AutoTrain](https://hf.co/docs/autotrain).
15
-
16
- # Usage
17
-
18
- ```python
19
-
20
- from transformers import AutoModelForCausalLM, AutoTokenizer
21
-
22
- model_path = "PATH_TO_THIS_REPO"
23
-
24
- tokenizer = AutoTokenizer.from_pretrained(model_path)
25
- model = AutoModelForCausalLM.from_pretrained(
26
- model_path,
27
- device_map="auto",
28
- torch_dtype='auto'
29
- ).eval()
30
-
31
- # Prompt content: "hi"
32
- messages = [
33
- {"role": "user", "content": "hi"}
34
- ]
35
-
36
- input_ids = tokenizer.apply_chat_template(conversation=messages, tokenize=True, add_generation_prompt=True, return_tensors='pt')
37
- output_ids = model.generate(input_ids.to('cuda'))
38
- response = tokenizer.decode(output_ids[0][input_ids.shape[1]:], skip_special_tokens=True)
39
-
40
- # Model response: "Hello! How can I assist you today?"
41
- print(response)
42
- ```
 
9
  pipeline_tag: text2text-generation
10
  ---
11
 
12
+ # ModeliCo - The Modelica GPT