transformers.pipeline:1from transformers import pipeline
2
3messages = [
4 {"role": "user", "content": "Write softmax formula in math style for me"},
5]
6pipe = pipeline("text-generation", model="explorewithai/LoxaPro", device_map = "cuda")
7pipe(messages)