diff --git a/requirements.txt b/requirements.txt
index ee04e43f..7f9d5982 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,7 +1,7 @@
 transformers
 openai
 langchain
-torch==2.0.0
+torch
 torchvision
 asyncio
 nest_asyncio
diff --git a/setup.py b/setup.py
index 01945f47..83647efa 100644
--- a/setup.py
+++ b/setup.py
@@ -4,7 +4,7 @@ from setuptools import setup, find_packages
 setup(
   name = 'swarms',
   packages = find_packages(exclude=[]),
-  version = '0.4.3',
+  version = '0.4.4',
   license='MIT',
   description = 'Swarms - Pytorch',
   author = 'Kye Gomez',
@@ -21,7 +21,7 @@ setup(
         'transformers',
         'openai',
         'langchain',
-        'torch==2.0.0',
+        'torch',
         'torchvision',
         'asyncio',
         'nest_asyncio',