Metadata-Version: 2.1
Name: potato-distillery
Version: 0.0.1
Summary: A package that distills large ML models.
Home-page: https://github.com/SpyrosMouselinos/potato_distillery
Author: Spyros Mouselinos
Author-email: s.mouselinos@uw.edu.pl
License: UNKNOWN
Platform: UNKNOWN
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Requires-Python: >=3.6
Description-Content-Type: text/markdown
License-File: LICENSE

# Potato Distillery 🥔 ➡  🥃

The purpose of this package is to make easier for users
to create smaller versions of popular ML models, in order
to fine-tune distilled versions of them.

Requirements:
* Huggingface 🤗
* Transformers 🤖
* Pytorch 🔥

Current support:
* BERT (Pytorch)
* GPT-2 (Pytorch)
* GPT-Neo (Pytorch)





