cogforge-engine 1.0.6__tar.gz → 1.0.7__tar.gz

{cogforge_engine-1.0.6 → cogforge_engine-1.0.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cogforge-engine
-Version: 1.0.6
+Version: 1.0.7
 Summary: A custom autograd engine and Transformer block built from scratch.
 Project-URL: Homepage, https://github.com/avikmjd2/cogforge
 Author-email: Avik Majumder <avikmjd2@gmail.com>

{cogforge_engine-1.0.6 → cogforge_engine-1.0.7}/cogforge/app.py

@@ -316,6 +316,9 @@ class Tensor:
         
         for node in reversed(topo):
             node._backwards()
+            #TRIAL
+            node._backwards = lambda: None
+            node._children.clear()
                     
             
     def flatten(self):

{cogforge_engine-1.0.6 → cogforge_engine-1.0.7}/cogforge/models.py

@@ -81,4 +81,5 @@ class GPT2:
             p = numpy.exp(z).astype(numpy.float64); p /= p.sum(-1, keepdims=True)
             nxt = numpy.array([[numpy.random.choice(len(pr), p=pr)] for pr in p])
             idx = numpy.concatenate([idx, nxt], axis=1)
+            # backend._cp.get_default_memory_pool().free_all_blocks() if backend.USE_GPU else None
         return idx                              

{cogforge_engine-1.0.6 → cogforge_engine-1.0.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "cogforge-engine" 
-version = "1.0.6"
+version = "1.0.7"
 authors = [
   { name="Avik Majumder", email="avikmjd2@gmail.com" },
 ]