Views
No views yet
SageBeluga13 model, hosted on Hugging Face, has been fine-tuned for specific tasks.1from transformers import AutoTokenizer, AutoModelForCausalLM
2import transformers
3import torch
4
5model = "davesoma/SageBeluga13"
6tokenizer = AutoTokenizer.from_pretrained(model)
7
8pipeline = transformers.pipeline(
9 "text-generation",
10 model=model,
11 tokenizer=tokenizer,
12 torch_dtype=torch.float32
13)
14
15sequences = pipeline(
16 "Girafatron is obsessed with giraffes...",
17 max_length=200,
18 do_sample=True,
19 top_k=10,
20 num_return_sequences=1,
21 eos_token_id=tokenizer.eos_token_id
22)
23
24for seq in sequences:
25 print(f"Result: {seq['generated_text']}")