0.9.4

ContextualAI · Feb 16, 2024 · 9f7f02c · 9f7f02c
1 parent 64b1d4c
commit 9f7f02c
Show file tree

Hide file tree

Showing 26 changed files with 5 additions and 2 deletions.
diff --git a/__init__.py b/__init__.py
diff --git a/gritlm/__init__.py b/gritlm/__init__.py
@@ -0,0 +1,3 @@
+__version__ = "0.9.4"
+
+from .gritlm import GritLM
diff --git a/gritlm.py → gritlm/gritlm.py b/gritlm.py → gritlm/gritlm.py
diff --git a/training/GradCache/.gitignore → gritlm/training/GradCache/.gitignore b/training/GradCache/.gitignore → gritlm/training/GradCache/.gitignore
diff --git a/training/GradCache/LICENSE → gritlm/training/GradCache/LICENSE b/training/GradCache/LICENSE → gritlm/training/GradCache/LICENSE
diff --git a/training/GradCache/README.md → gritlm/training/GradCache/README.md b/training/GradCache/README.md → gritlm/training/GradCache/README.md
diff --git a/training/GradCache/setup.py → gritlm/training/GradCache/setup.py b/training/GradCache/setup.py → gritlm/training/GradCache/setup.py
diff --git a/...ning/GradCache/src/grad_cache/__init__.py → ...ning/GradCache/src/grad_cache/__init__.py b/...ning/GradCache/src/grad_cache/__init__.py → ...ning/GradCache/src/grad_cache/__init__.py
diff --git a/...adCache/src/grad_cache/cachex/__init__.py → ...adCache/src/grad_cache/cachex/__init__.py b/...adCache/src/grad_cache/cachex/__init__.py → ...adCache/src/grad_cache/cachex/__init__.py
diff --git a/...Cache/src/grad_cache/cachex/functional.py → ...Cache/src/grad_cache/cachex/functional.py b/...Cache/src/grad_cache/cachex/functional.py → ...Cache/src/grad_cache/cachex/functional.py
diff --git a/...adCache/src/grad_cache/cachex/training.py → ...adCache/src/grad_cache/cachex/training.py b/...adCache/src/grad_cache/cachex/training.py → ...adCache/src/grad_cache/cachex/training.py
diff --git a/...Cache/src/grad_cache/cachex/tree_utils.py → ...Cache/src/grad_cache/cachex/tree_utils.py b/...Cache/src/grad_cache/cachex/tree_utils.py → ...Cache/src/grad_cache/cachex/tree_utils.py
diff --git a/...dCache/src/grad_cache/context_managers.py → ...dCache/src/grad_cache/context_managers.py b/...dCache/src/grad_cache/context_managers.py → ...dCache/src/grad_cache/context_managers.py
diff --git a/...ng/GradCache/src/grad_cache/functional.py → ...ng/GradCache/src/grad_cache/functional.py b/...ng/GradCache/src/grad_cache/functional.py → ...ng/GradCache/src/grad_cache/functional.py
diff --git a/...ng/GradCache/src/grad_cache/grad_cache.py → ...ng/GradCache/src/grad_cache/grad_cache.py b/...ng/GradCache/src/grad_cache/grad_cache.py → ...ng/GradCache/src/grad_cache/grad_cache.py
diff --git a/training/GradCache/src/grad_cache/loss.py → ...training/GradCache/src/grad_cache/loss.py b/training/GradCache/src/grad_cache/loss.py → ...training/GradCache/src/grad_cache/loss.py
diff --git a/training/arguments.py → gritlm/training/arguments.py b/training/arguments.py → gritlm/training/arguments.py
diff --git a/training/data.py → gritlm/training/data.py b/training/data.py → gritlm/training/data.py
diff --git a/training/gradcache_trainer.py → gritlm/training/gradcache_trainer.py b/training/gradcache_trainer.py → gritlm/training/gradcache_trainer.py
diff --git a/training/model.py → gritlm/training/model.py b/training/model.py → gritlm/training/model.py
diff --git a/training/run.py → gritlm/training/run.py b/training/run.py → gritlm/training/run.py
diff --git a/training/toy_data/toy_data_embedding.jsonl → ...raining/toy_data/toy_data_embedding.jsonl b/training/toy_data/toy_data_embedding.jsonl → ...raining/toy_data/toy_data_embedding.jsonl
diff --git a/training/toy_data/toy_data_generative.jsonl → ...aining/toy_data/toy_data_generative.jsonl b/training/toy_data/toy_data_generative.jsonl → ...aining/toy_data/toy_data_generative.jsonl
diff --git a/...oy_data_instruct/toy_data_embedding.jsonl → ...oy_data_instruct/toy_data_embedding.jsonl b/...oy_data_instruct/toy_data_embedding.jsonl → ...oy_data_instruct/toy_data_embedding.jsonl
diff --git a/...y_data_instruct/toy_data_generative.jsonl → ...y_data_instruct/toy_data_generative.jsonl b/...y_data_instruct/toy_data_generative.jsonl → ...y_data_instruct/toy_data_generative.jsonl
diff --git 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=[
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,3 @@
		__version__ = "0.9.4"

		from .gritlm import GritLM