mosaicml

/

mpt-30b-instruct

like 101

Follow

Mosaic ML, Inc. 360

Hi Team, I was trying instruct model for summarization tasks but in multiple cases I observed that at the end of summarized passage its appending some code or arbitrary words like below:

period.package com.github.tix320.skimp.api.level.entity;\n\nimport com.github.tix320.skimp.api.level.SkimpLevel;\nimport com.github.tix320.skimp.api.util.Tuple;\nimport com.github.tix320.skimp.api.world.World;\nimport com.github.tix320.skimp.api.world.

#ifndef LIB_UTIL_H\n#define LIB_UTIL_H\n\n#include \n#include \n#include \n#include \n#include \n#include \n#include \n#include \n#include \n#include \n\n#include <Eigen/Core>\n\n#define UNUSED(x) ((void)(x))

Just FYI using below code:

import torch
import transformers
from transformers import pipeline
import pandas as pd

name = 'mosaicml/mpt-30b-instruct'

config = transformers.AutoConfig.from_pretrained(name, trust_remote_code=True)
config.attn_config['attn_impl'] = 'triton'
config.init_device = 'cuda:0'

tokenizer = transformers.AutoTokenizer.from_pretrained("mosaicml/mpt-30b")
model = transformers.AutoModelForCausalLM.from_pretrained(
name,
config=config,
torch_dtype=torch.bfloat16, # Load model weights in bfloat16
trust_remote_code=True
)

def format_prompt(instruction):
template = "Below is an instruction that describes a task. Write a response that appropriately completes the request.\n\n###Instruction\n{instruction}\n\n### Response\n"
return template.format(instruction=instruction)

def summarize(text):
text = 'Please summarize the following article. Only highlighting the important points. \n\n' + text
fmt_ex = format_prompt(instruction=text)
with torch.autocast('cuda', dtype=torch.bfloat16):
inputs = tokenizer(fmt_ex, return_tensors="pt").to('cuda:0')
outputs = model.generate(**inputs, max_new_tokens=256, do_sample=True, top_p=0.95, top_k=50, temperature=0.7)
summary = tokenizer.batch_decode(outputs, skip_special_tokens=True)
summary = summary[0].split("### Response")[1]
return summary

I tried to play around with the generate parameters but not much help