diff --git a/__init__.py b/__init__.py deleted file mode 100644 index c598173..0000000 --- a/__init__.py +++ /dev/null @@ -1 +0,0 @@ -__version__ = "0.9.3" diff --git a/gritlm/__init__.py b/gritlm/__init__.py new file mode 100644 index 0000000..47bde56 --- /dev/null +++ b/gritlm/__init__.py @@ -0,0 +1,3 @@ +__version__ = "0.9.4" + +from .gritlm import GritLM \ No newline at end of file diff --git a/gritlm.py b/gritlm/gritlm.py similarity index 100% rename from gritlm.py rename to gritlm/gritlm.py diff --git a/training/GradCache/.gitignore b/gritlm/training/GradCache/.gitignore similarity index 100% rename from training/GradCache/.gitignore rename to gritlm/training/GradCache/.gitignore diff --git a/training/GradCache/LICENSE b/gritlm/training/GradCache/LICENSE similarity index 100% rename from training/GradCache/LICENSE rename to gritlm/training/GradCache/LICENSE diff --git a/training/GradCache/README.md b/gritlm/training/GradCache/README.md similarity index 100% rename from training/GradCache/README.md rename to gritlm/training/GradCache/README.md diff --git a/training/GradCache/setup.py b/gritlm/training/GradCache/setup.py similarity index 100% rename from training/GradCache/setup.py rename to gritlm/training/GradCache/setup.py diff --git a/training/GradCache/src/grad_cache/__init__.py b/gritlm/training/GradCache/src/grad_cache/__init__.py similarity index 100% rename from training/GradCache/src/grad_cache/__init__.py rename to gritlm/training/GradCache/src/grad_cache/__init__.py diff --git a/training/GradCache/src/grad_cache/cachex/__init__.py b/gritlm/training/GradCache/src/grad_cache/cachex/__init__.py similarity index 100% rename from training/GradCache/src/grad_cache/cachex/__init__.py rename to gritlm/training/GradCache/src/grad_cache/cachex/__init__.py diff --git a/training/GradCache/src/grad_cache/cachex/functional.py b/gritlm/training/GradCache/src/grad_cache/cachex/functional.py similarity index 100% rename from training/GradCache/src/grad_cache/cachex/functional.py rename to gritlm/training/GradCache/src/grad_cache/cachex/functional.py diff --git a/training/GradCache/src/grad_cache/cachex/training.py b/gritlm/training/GradCache/src/grad_cache/cachex/training.py similarity index 100% rename from training/GradCache/src/grad_cache/cachex/training.py rename to gritlm/training/GradCache/src/grad_cache/cachex/training.py diff --git a/training/GradCache/src/grad_cache/cachex/tree_utils.py b/gritlm/training/GradCache/src/grad_cache/cachex/tree_utils.py similarity index 100% rename from training/GradCache/src/grad_cache/cachex/tree_utils.py rename to gritlm/training/GradCache/src/grad_cache/cachex/tree_utils.py diff --git a/training/GradCache/src/grad_cache/context_managers.py b/gritlm/training/GradCache/src/grad_cache/context_managers.py similarity index 100% rename from training/GradCache/src/grad_cache/context_managers.py rename to gritlm/training/GradCache/src/grad_cache/context_managers.py diff --git a/training/GradCache/src/grad_cache/functional.py b/gritlm/training/GradCache/src/grad_cache/functional.py similarity index 100% rename from training/GradCache/src/grad_cache/functional.py rename to gritlm/training/GradCache/src/grad_cache/functional.py diff --git a/training/GradCache/src/grad_cache/grad_cache.py b/gritlm/training/GradCache/src/grad_cache/grad_cache.py similarity index 100% rename from training/GradCache/src/grad_cache/grad_cache.py rename to gritlm/training/GradCache/src/grad_cache/grad_cache.py diff --git a/training/GradCache/src/grad_cache/loss.py b/gritlm/training/GradCache/src/grad_cache/loss.py similarity index 100% rename from training/GradCache/src/grad_cache/loss.py rename to gritlm/training/GradCache/src/grad_cache/loss.py diff --git a/training/arguments.py b/gritlm/training/arguments.py similarity index 100% rename from training/arguments.py rename to gritlm/training/arguments.py diff --git a/training/data.py b/gritlm/training/data.py similarity index 100% rename from training/data.py rename to gritlm/training/data.py diff --git a/training/gradcache_trainer.py b/gritlm/training/gradcache_trainer.py similarity index 100% rename from training/gradcache_trainer.py rename to gritlm/training/gradcache_trainer.py diff --git a/training/model.py b/gritlm/training/model.py similarity index 100% rename from training/model.py rename to gritlm/training/model.py diff --git a/training/run.py b/gritlm/training/run.py similarity index 100% rename from training/run.py rename to gritlm/training/run.py diff --git a/training/toy_data/toy_data_embedding.jsonl b/gritlm/training/toy_data/toy_data_embedding.jsonl similarity index 100% rename from training/toy_data/toy_data_embedding.jsonl rename to gritlm/training/toy_data/toy_data_embedding.jsonl diff --git a/training/toy_data/toy_data_generative.jsonl b/gritlm/training/toy_data/toy_data_generative.jsonl similarity index 100% rename from training/toy_data/toy_data_generative.jsonl rename to gritlm/training/toy_data/toy_data_generative.jsonl diff --git a/training/toy_data_instruct/toy_data_embedding.jsonl b/gritlm/training/toy_data_instruct/toy_data_embedding.jsonl similarity index 100% rename from training/toy_data_instruct/toy_data_embedding.jsonl rename to gritlm/training/toy_data_instruct/toy_data_embedding.jsonl diff --git a/training/toy_data_instruct/toy_data_generative.jsonl b/gritlm/training/toy_data_instruct/toy_data_generative.jsonl similarity index 100% rename from training/toy_data_instruct/toy_data_generative.jsonl rename to gritlm/training/toy_data_instruct/toy_data_generative.jsonl diff --git a/setup.py b/setup.py index 17f2cdc..f624f60 100644 --- a/setup.py +++ b/setup.py @@ -43,7 +43,7 @@ setup( name='gritlm', - version='0.9.3', + version='0.9.4', description='GritLM', long_description=readme, long_description_content_type="text/markdown", @@ -55,6 +55,7 @@ "Huggingface Organization": "https://huggingface.co/gritlm", "Source Code": "https://github.com/ContextualAI/gritlm", }, + url="https://github.com/ContextualAI/gritlm", packages=find_packages(), python_requires=">=3.7.0", install_requires=[