Views
No views yet
1from transformers import pipeline
2import torch
3prompt= "FILL IN THE QUESTION"
4generator = pipeline(
5 model="HaVen-CodeQwen",
6 task="text-generation",
7 torch_dtype=torch.bfloat16,
8 device_map="auto",
9)
10result = generator(prompt , max_length=2048,num_return_sequences=1, temperature=0.0)
11response = result[0]["generated_text"]
12print("Response:", response)