statedict2pytree 0.5.3__tar.gz → 0.5.4__tar.gz

{statedict2pytree-0.5.3 → statedict2pytree-0.5.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: statedict2pytree
-Version: 0.5.3
+Version: 0.5.4
 Summary: Converts torch models into PyTrees for Equinox
 Author-email: "Artur A. Galstyan" <mail@arturgalstyan.dev>
 Requires-Python: ~=3.10
@@ -15,6 +15,7 @@ Requires-Dist: jaxtyping
 Requires-Dist: loguru
 Requires-Dist: penzai
 Requires-Dist: pydantic
+Requires-Dist: pytest
 Requires-Dist: python-dotenv
 Requires-Dist: torch
 Requires-Dist: torchvision
@@ -32,6 +33,11 @@ Description-Content-Type: text/markdown
 
 ![statedict2pytree](statedict2pytree.png "A ResNet demo")
 
+## Docs
+
+Docs can be found [here](https://artur-galstyan.github.io/statedict2pytree/).
+
+
 ## Important
 
 This package is still in its infancy and hihgly experimental! The code works, but it's far from perfect. With more and more iterations, it will eventually become stable and well tested.
@@ -67,9 +73,6 @@ Currently, there is no sophisticated shape matching in place. Two matrices are c
 (8, 1, 1) and (8, ) match, because (8 _ 1 _ 1 = 8)
 
 
-### Docs
-
-Documentation will appear as soon as I have all the necessary features implemented. Until then, check out the "main.py" file for a better example.
 
 ### Disclaimer
 

{statedict2pytree-0.5.3 → statedict2pytree-0.5.4}/README.md

@@ -2,6 +2,11 @@
 
 ![statedict2pytree](statedict2pytree.png "A ResNet demo")
 
+## Docs
+
+Docs can be found [here](https://artur-galstyan.github.io/statedict2pytree/).
+
+
 ## Important
 
 This package is still in its infancy and hihgly experimental! The code works, but it's far from perfect. With more and more iterations, it will eventually become stable and well tested.
@@ -37,9 +42,6 @@ Currently, there is no sophisticated shape matching in place. Two matrices are c
 (8, 1, 1) and (8, ) match, because (8 _ 1 _ 1 = 8)
 
 
-### Docs
-
-Documentation will appear as soon as I have all the necessary features implemented. Until then, check out the "main.py" file for a better example.
 
 ### Disclaimer
 

{statedict2pytree-0.5.3 → statedict2pytree-0.5.4}/client/src/App.svelte

@@ -5,7 +5,7 @@
     import Swal from "sweetalert2";
 
     let model: string = "model.eqx";
-    let anthropicModel: "opus" | "sonnet" | "haiku" = "haiku";
+    let anthropicModel: "opus" | "sonnet" | "sonnet3.5" | "haiku" = "haiku";
 
     const Toast = Swal.mixin({
         toast: true,
@@ -492,6 +492,7 @@ Here's your input:
         <select bind:value={anthropicModel}>
             <option value="opus">opus</option>
             <option value="sonnet">sonnet</option>
+            <option value="sonnet3.5">sonnet3.5</option>
             <option value="haiku">haiku</option>
         </select>
     </div>

{statedict2pytree-0.5.3 → statedict2pytree-0.5.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "statedict2pytree"
-version = "0.5.3"
+version = "0.5.4"
 description = "Converts torch models into PyTrees for Equinox"
 readme = "README.md"
 requires-python = "~=3.10"
@@ -21,6 +21,7 @@ dependencies = [
     "jaxonmodels",
     "anthropic",
     "python-dotenv",
+    "pytest",
 ]
 [project.optional-dependencies]
 dev = ["nox", "pre-commit", "pytest", "mkdocs"]