def METHOD_NAME(self):
    model = AutoModelForCausalLM.from_pretrained(
        "stabilityai/stablelm-tuned-alpha-{}b".format(
            self.config.stabilityai.model_size
        ),
        use_auth_token=self.config.stabilityai.api_key,
        torch_dtype=torch.float16,
    ).cuda()
    return pipeline(
        "text-generation",
        model,
        tokenizer=self.tokenizer,
        device=0,
        max_new_tokens=self.config.stabilityai.max_new_tokens,
        num_return_sequences=self.config.stabilityai.num_return_sequences,
        num_beams=self.config.stabilityai.num_beams,
        do_sample=self.config.stabilityai.do_sample,
        temperature=self.config.stabilityai.temperature,
        top_p=self.config.stabilityai.top_p,
        top_k=self.config.stabilityai.top_k,
        stopping_criteria=StoppingCriteriaList([StopOnTokens()]),
    )