Views
No views yet
1from unsloth import FastLanguageModel
2import torch
3max_seq_length = 4096
4dtype = torch.float16
5load_in_4bit = True
6
7model, tokenizer = FastLanguageModel.from_pretrained(
8 model_name = "ArvindSharma18/Phi-3-mini-4k-instruct-bnb-4bit-Clinical-Trail-Merged",
9 max_seq_length = max_seq_length,
10 dtype = dtype,
11 load_in_4bit = load_in_4bit
12)
13FastLanguageModel.for_inference(model)
14inputs = tokenizer(
15[
16 "Official Title: Randomized Trial of Usual Care vs. Specialized, Phase-specific Care for Youth at Risk for Psychosis"
17], return_tensors = "pt").to("cuda")
18from transformers import TextStreamer
19text_streamer = TextStreamer(tokenizer, skip_prompt = True)
20_ = model.generate(input_ids = inputs.input_ids, attention_mask = inputs.attention_mask,
21 streamer = text_streamer, max_new_tokens = 2048, pad_token_id = tokenizer.eos_token_id)