duonlabs
/

apogee-nano-2.4M-base

state-space-models

Model card Files Files and versions

Theob commited on Mar 4

Commit

4702f0f

·

verified ·

1 Parent(s): 89b4b0c

Update handler.py

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -10,7 +10,7 @@ from pathlib import Path
 torch.backends.cuda.matmul.allow_tf32 = True # allow tf32 on matmul
 torch.backends.cudnn.allow_tf32 = True # allow tf32 on cudnn
-class ApogeeHandler:
     """
     Handler class.
     """
@@ -124,7 +124,7 @@ class ApogeeHandler:
 if __name__ == "__main__":
     import pandas as pd
-    handler = ApogeeHandler()
     test_path = Path(__file__).parents[2] / "tests" / "assets" / "BTCUSDT-1m-2019-03.csv"
     with open(test_path, "r") as f:
         data = pd.read_csv(f)

 torch.backends.cuda.matmul.allow_tf32 = True # allow tf32 on matmul
 torch.backends.cudnn.allow_tf32 = True # allow tf32 on cudnn
+class EndpointHandler:
     """
     Handler class.
     """
 if __name__ == "__main__":
     import pandas as pd
+    handler = EndpointHandler()
     test_path = Path(__file__).parents[2] / "tests" / "assets" / "BTCUSDT-1m-2019-03.csv"
     with open(test_path, "r") as f:
         data = pd.read_csv(f)