ArvindSharma18's picture
Update README.md
ed966df verified
metadata
base_model: unsloth/Phi-3-mini-4k-instruct-bnb-4bit
language:
  - en
license: creativeml-openrail-m
tags:
  - text-generation-inference
  - transformers
  - unsloth
  - trl
  - sft

Note

How to Use

Note: EOS issue. Will keep on Generating Secondary Outcomes.

from unsloth import FastLanguageModel
import torch
max_seq_length = 4096 
dtype = torch.float16
load_in_4bit = True

model, tokenizer = FastLanguageModel.from_pretrained(
    model_name = "ArvindSharma18/Phi-3-mini-4k-instruct-bnb-4bit-Clinical-Trail-Merged",
    max_seq_length = max_seq_length,
    dtype = dtype,
    load_in_4bit = load_in_4bit
)
FastLanguageModel.for_inference(model)
inputs = tokenizer(
[
    "Official Title: Randomized Trial of Usual Care vs. Specialized, Phase-specific Care for Youth at Risk for Psychosis"
], return_tensors = "pt").to("cuda")
from transformers import TextStreamer
text_streamer = TextStreamer(tokenizer, skip_prompt = True)
_ = model.generate(input_ids = inputs.input_ids, attention_mask = inputs.attention_mask,
                   streamer = text_streamer, max_new_tokens = 2048, pad_token_id = tokenizer.eos_token_id)

Uploaded model

  • Developed by: ArvindSharma18
  • Finetuned from model : unsloth/Phi-3-mini-4k-instruct-bnb-4bit

This model was trained 2x faster with Unsloth and Huggingface's TRL library.