Views
No views yet
pip install transformers==4.35.21import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4tokenizer = AutoTokenizer.from_pretrained("Upstage/SOLAR-10.7B-Instruct-v1.0")
5model = AutoModelForCausalLM.from_pretrained(
6 "Upstage/SOLAR-10.7B-Instruct-v1.0",
7 device_map="auto",
8 torch_dtype=torch.float16,
9)1conversation = [ {'role': 'user', 'content': 'Hello?'} ]
2
3prompt = tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt=True)
4
5inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
6outputs = model.generate(**inputs, use_cache=True, max_length=4096)
7output_text = tokenizer.decode(outputs[0])
8print(output_text)<s> ### User:
Hello?
### Assistant:
Hello, how can I assist you today? Please feel free to ask any questions or request help with a specific task.</s>