ruixie commited on
Commit
f5422a8
1 Parent(s): 3ce2da1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -4
README.md CHANGED
@@ -40,8 +40,8 @@ Codeshell offers a model in the Hugging Face format. Developers can load and use
40
  ```python
41
  import torch
42
  from transformers import AutoModelForCausalLM, AutoTokenizer
43
- tokenizer = AutoTokenizer.from_pretrained("codeshell", trust_remote_code=True)
44
- model = AutoModelForCausalLM.from_pretrained("codeshell", trust_remote_code=True).cuda()
45
  inputs = tokenizer('def print_hello_world():', return_tensors='pt').cuda()
46
  outputs = model.generate(inputs)
47
  print(tokenizer.decode(outputs[0]))
@@ -123,8 +123,6 @@ Despite our significant efforts to ensure compliance in the data used during the
123
 
124
  在满足上述条件的前提下,您需要通过向codeshell.opensource@gmail.com发送电子邮件提交申请。经审核通过后,将授予您一个全球的、非排他的、不可转让的、不可再授权的商业版权许可。
125
 
126
-
127
-
128
  Community use of the CodeShell model requires adherence to the CodeShell Model License Agreement and the Apache 2.0 License. The CodeShell model is allowed for commercial use, but if you plan to use the CodeShell model or its derivatives for commercial purposes, you need to ensure that the entity meets the following conditions:
129
 
130
  1. The Daily Active Users (DAU) of your or your affiliate's service or product is less than 1 million.
 
40
  ```python
41
  import torch
42
  from transformers import AutoModelForCausalLM, AutoTokenizer
43
+ tokenizer = AutoTokenizer.from_pretrained("WisdomShell/CodeShell-7B", trust_remote_code=True)
44
+ model = AutoModelForCausalLM.from_pretrained("WisdomShell/CodeShell-7B", trust_remote_code=True).cuda()
45
  inputs = tokenizer('def print_hello_world():', return_tensors='pt').cuda()
46
  outputs = model.generate(inputs)
47
  print(tokenizer.decode(outputs[0]))
 
123
 
124
  在满足上述条件的前提下,您需要通过向codeshell.opensource@gmail.com发送电子邮件提交申请。经审核通过后,将授予您一个全球的、非排他的、不可转让的、不可再授权的商业版权许可。
125
 
 
 
126
  Community use of the CodeShell model requires adherence to the CodeShell Model License Agreement and the Apache 2.0 License. The CodeShell model is allowed for commercial use, but if you plan to use the CodeShell model or its derivatives for commercial purposes, you need to ensure that the entity meets the following conditions:
127
 
128
  1. The Daily Active Users (DAU) of your or your affiliate's service or product is less than 1 million.