azimjon commited on
Commit
e182078
1 Parent(s): 64971b2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -97,7 +97,7 @@ messages = [
97
  {"role": "system", "content": "You are a pirate chatbot who always responds in pirate speak!"},
98
  {"role": "user", "content": "Who are you?"},
99
  ]
100
- chatbot = pipeline("text-generation", model="behbudiy/Mistral-7B-Instruct-Uz")
101
  chatbot(messages)
102
  ```
103
 
 
97
  {"role": "system", "content": "You are a pirate chatbot who always responds in pirate speak!"},
98
  {"role": "user", "content": "Who are you?"},
99
  ]
100
+ chatbot = pipeline("text-generation", model="behbudiy/Mistral-7B-Instruct-Uz", device='cuda')
101
  chatbot(messages)
102
  ```
103