Views
No views yet
LlamaTokenizer class, or passing in the use_fast=False option for the AutoTokenizer class. See the following example for usage.1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3tokenizer = AutoTokenizer.from_pretrained("PengQu/open_llama_7b_v2_vicuna_Chinese",use_fast=False)
4model = AutoModelForCausalLM.from_pretrained("PengQu/open_llama_7b_v2_vicuna_Chinese").to("cuda")
5
6instruction = "A chat between a curious user and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the user's questions. USER: {} ASSISTANT:"
7prompt = instruction.format('用flask写一个简单的http服务器。')
8input_ids = tokenizer(prompt, return_tensors="pt").input_ids.to("cuda")
9
10generation_output = model.generate(input_ids=input_ids, max_new_tokens=512)
11print(tokenizer.decode(generation_output[0],skip_special_tokens=True))用flask写一个简单的http服务器。
from flask import Flask
app = Flask(__name__)
@app.route('/')
def hello():
return 'Hello, World!'
if __name__ == '__main__':
app.run()
这段代码定义了一个Flask应用程序,并为根路径('/')定义了一个路由。当用户在其Web浏览器中导航到该路径时,将调用`hello()`函数,并返回字符串“Hello, World!”。
要运行此代码,您需要在计算机上安装Flask。您可以使用以下命令使用pip安装它:
pip install Flask
安装Flask后,您可以使用以下命令运行代码:
python app.py
这将启动一个本地开发服务器,您可以使用Web浏览器访问它,方法是导航到`http://localhost:5000/`。
您还可以通过添加其他路由和功能来进一步自定义代码。例如,您可以为不同的端点定义不同的路由,并使用请求数据执行某些操作。您还可以向应用程序添加错误处理和用户身份验证。