Merge pull request #135 from helicalAI/main

Fix version

examples/notebooks/Helix-mRNA.ipynb

@@ -38,9 +38,10 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "!pip install torch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 --index-url https://download.pytorch.org/whl/cu121\n",
-    "!pip uninstall mamba-ssm causal-conv1d\n",
-    "!pip install causal-conv1d==1.4.0 && pip install mamba-ssm==2.2.2"
+    "!pip uninstall torch torchtext torchaudio -y\n",
+    "!pip install torch==2.3.0 torchtext torchaudio --index-url https://download.pytorch.org/whl/cu121\n",
+    "!pip uninstall mamba-ssm causal-conv1d -y\n",
+    "!pip install mamba-ssm==2.2.2 --no-cache-dir"
    ]
   },
   {
@@ -49,7 +50,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "!pip install helical"
+    "!pip install helical\n",
+    "!pip uninstall causal-conv1d -y"
    ]
   },
   {
@@ -79,8 +81,7 @@
     "import torch\n",
     "import pandas as pd\n",
     "\n",
-    "#  Setting to CPU by default because of colab issues, if not run on colab, set to \"cuda\"\n",
-    "device=\"cpu\""
+    "device = \"cuda\" if torch.cuda.is_available() else \"cpu\""
    ]
   },
   {

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "helical"
-version = "0.0.1a8"
+version = "0.0.1a9"
 authors = [
   { name="Helical Team", email="support@helical-ai.com" },
 ]
@@ -46,7 +46,7 @@ dependencies = [
     'pyensembl',
     'datasets==2.20.0',
     'mamba-ssm==2.2.2',
-    'causal-conv1d==1.4.0'
+    'causal-conv1d==1.4.0',
 ]