sdk

SubstrateLabs · Aug 14, 2024 · d06c388 · d06c388
1 parent b92884b
commit d06c388
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/substrate/__init__.py b/substrate/__init__.py
@@ -17,6 +17,7 @@
     Experimental,
     FetchVectors,
     Firellava13B,
+    GenerateCode,
     InpaintImage,
     UpscaleImage,
     DeleteVectors,
@@ -71,6 +72,7 @@
     "BatchComputeJSON",
     "ComputeJSON",
     "MultiComputeJSON",
+    "GenerateCode",
     "Mistral7BInstruct",
     "Mixtral8x7BInstruct",
     "Llama3Instruct8B",

diff --git a/substrate/nodes.py b/substrate/nodes.py
@@ -26,6 +26,7 @@
         ExperimentalOut,
         FetchVectorsOut,
         Firellava13BOut,
+        GenerateCodeOut,
         InpaintImageOut,
         UpscaleImageOut,
         DeleteVectorsOut,
@@ -77,6 +78,7 @@
     FutureExperimentalOut,
     FutureFetchVectorsOut,
     FutureFirellava13BOut,
+    FutureGenerateCodeOut,
     FutureInpaintImageOut,
     FutureUpscaleImageOut,
     FutureDeleteVectorsOut,