View file src/colab/solar_10_7b.py - Download
# -*- coding: utf-8 -*-
"""SOLAR-10.7B.ipynb
Automatically generated by Colaboratory.
Original file is located at
https://colab.research.google.com/drive/1N_f-7KXp_dctLLkiMqfELTpnF3psoY3J
https://www.atpostlight.com/?_=%2Fupstage%2FSOLAR-10.7B-v1.0%23KJWqMdlUlBnjPuoSWRPngYr2fc9jFA%3D%3D
"""
!pip install accelerate
import torch
from transformers import AutoModelForCausalLM, AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained("Upstage/SOLAR-10.7B-v1.0")
model = AutoModelForCausalLM.from_pretrained(
"Upstage/SOLAR-10.7B-v1.0",
device_map="auto",
torch_dtype=torch.float16,
)
text = "Hi, my name is "
inputs = tokenizer(text, return_tensors="pt")
outputs = model.generate(**inputs, max_new_tokens=64)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))