Views
No views yet
from transformers import AutoModel, AutoConfig, AutoModelForCausalLM, AutoTokenizer
# Path to the directory containing the model files
model_directory = "autopilot-ai/Indic-sentence-completion"
tokenizer = AutoTokenizer.from_pretrained(model_directory)
model = AutoModelForCausalLM.from_pretrained(
model_directory,
load_in_8bit=True,
device_map="auto",
)
# Load the model configuration
config = AutoConfig.from_pretrained(model_directory)
# Load the model
model = AutoModel.from_pretrained(model_directory, config=config)
batch = tokenizer("હેલો કેમ છો?", return_tensors='pt')
with torch.cuda.amp.autocast():
output_tokens = model.generate(**batch, max_new_tokens=10)
print('\n\n', tokenizer.decode(output_tokens[0], skip_special_tokens=True))pip install accelerate bitsandbytes