Views
No views yet

| Industry | Version | Llama 3.1 8B |
|---|---|---|
| Food And Beverage | bf16 | ContaAI/ContaLLM-Food-Beverage-8B-Instruct |
| Food And Beverage | 8bit | ContaAI/ContaLLM-Food-Beverage-8B-Instruct-8bit |
| Food And Beverage | 4bit | ContaAI/ContaLLM-Food-Beverage-8B-Instruct-4bit |
from transformers import AutoModelForCausalLM
model = AutoModelForCausalLM.from_pretrained("ContaAI/ContaLLM-Food-Beverage-8B-Instruct-8bit")system_prompt = '请根据用户提供的营销需求、选品及其他信息写一篇食品饮料行业的营销推文。'| Parameter name | Required | Meaning and optional range |
|---|---|---|
| 营销需求 | required | Fill in your marketing requirements, cannot be blank |
| 选品 | required | Fill in your product selection, cannot be blank |
| 选品知识库 | required | Fill in the relevant information/materials about your product, cannot be blank |
| 关键词 | optional | Fill in your marketing keywords, or remove this row from the prompt |
| 标签 | optional | Fill in the hashtag, or remove this row from the prompt |
| 主推卖点 | optional | Fill in the main recommended selling points, or remove this row from the prompt |
| 主推场景 | optional | Fill in the main recommended scenes, or remove this row from the prompt |
| 文章类型 | optional | Fill in the article type, or remove this row from the prompt |
user_prompt = """营销需求:夏日清凉,日料风味体验
选品:清新柠檬寿司卷
选品知识库:1、选用新鲜的三文鱼和牛油果,搭配清爽柠檬汁,口感层次丰富。2、低脂健康,适合健身人士。3、每份仅含200大卡,轻松享受美味。
关键词:日料、寿司、健康饮食、夏日美食
主推卖点:清新健康
主推场景:夏日聚会
标签:#日料# #寿司# #健康美食
文章类型:美食推荐"""import torch
from transformers import AutoModelForCausalLM, AutoTokenizer
model_name = "ContaAI/ContaLLM-Food-Beverage-8B-Instruct-8bit"
model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
tokenizer = AutoTokenizer.from_pretrained(model_name)
system_prompt = '请根据用户提供的营销需求、选品及其他信息写一篇食品饮料行业的营销推文。'
user_prompt = """营销需求:夏日清凉,日料风味体验
选品:清新柠檬寿司卷
选品知识库:1、选用新鲜的三文鱼和牛油果,搭配清爽柠檬汁,口感层次丰富。2、低脂健康,适合健身人士。3、每份仅含200大卡,轻松享受美味。
关键词:日料、寿司、健康饮食、夏日美食
主推卖点:清新健康
主推场景:夏日聚会
标签:#日料# #寿司# #健康美食
文章类型:美食推荐"""
prompt_template = '''<|begin_of_text|><|start_header_id|>system<|end_header_id|>
{}<|eot_id|><|start_header_id|>user<|end_header_id|>
{}<|eot_id|><|start_header_id|>assistant<|end_header_id|>'''
prompt = prompt_template.format(system_prompt, user_prompt)
tokenized_message = tokenizer(
prompt,
max_length=2048,
return_tensors="pt",
add_special_tokens=False
)
response_token_ids= model.generate(
**tokenized_message,
max_new_tokens=1024,
do_sample=True,
top_p=1.0,
temperature=0.5,
min_length=None,
use_cache=True,
top_k=50,
repetition_penalty=1.2,
length_penalty=1,
)
generated_tokens = response_token_ids[0, tokenized_message['input_ids'].shape[-1]:]
generated_text = tokenizer.decode(generated_tokens, skip_special_tokens=True)
print(generated_text)