From 4bb33eb42fdb75f0e0f3a30f48b0200556dc9fc2 Mon Sep 17 00:00:00 2001
From: Eternal Reclaimer <98760976+kyegomez@users.noreply.github.com>
Date: Tue, 26 Mar 2024 10:49:03 -0700
Subject: [PATCH] Update ein_fft.py

---
 simba_torch/ein_fft.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/simba_torch/ein_fft.py b/simba_torch/ein_fft.py
index 067574d..936cff0 100644
--- a/simba_torch/ein_fft.py
+++ b/simba_torch/ein_fft.py
@@ -195,11 +195,11 @@ def forward(self, x: Tensor) -> Tensor:
 
 
 # Random input tensor
-x = torch.randn(1, 3, 64)
+# x = torch.randn(1, 3, 64)
 
-# Instantiate EinFFT module
-einfft = EinFFTText(3, 64)
+# # Instantiate EinFFT module
+# einfft = EinFFTText(3, 64)
 
-# Apply EinFFT to get an output
-output = einfft(x)
-print(output.shape)
+# # Apply EinFFT to get an output
+# output = einfft(x)
+# print(output.shape)