Funnybubble / app.py
Funybubble's picture
Update app.py
ffb307e verified
raw
history blame contribute delete
380 Bytes
from transformers import AutoModel
# Replace 'Funybubble/replit-code-instruct-quanatized' with the actual model name
model_name = 'Funybubble/replit-code-instruct-quanatized'
# Load the tokenizer and model
model = AutoModel.from_pretrained(model_name)
# Save the tokenizer and model to the same directory as app.py
save_directory = './'
model.save_pretrained(save_directory)