Views
No views yet
pip install -U transformerspipeline API1import torch
2from transformers import pipeline
3
4pipe = pipeline(
5 "text-generation",
6 model="petkopetkov/gemma-2-2b-bg",
7 torch_dtype=torch.bfloat16,
8 device_map="auto"
9)
10
11prompt = "Колко е 2 + 2?"
12
13print(pipe(prompt)[0]['generated_text'])