Use new Runtime Torch installer mechanism and update torch

BrokenSource · Mar 17, 2024 · 658483a · 658483a
1 parent 15fbefd
commit 658483a
Show file tree

Hide file tree

Showing 3 changed files with 130 additions and 559 deletions.
diff --git a/DepthFlow/__init__.py b/DepthFlow/__init__.py
@@ -7,8 +7,6 @@
 _spinner.start()
 
 import DepthFlow.Resources as DepthFlowResources
-import torch
-import transformers
 
 DEPTHFLOW = BrokenProject(
     PACKAGE=__file__,
@@ -19,7 +17,12 @@
 
 Broken.PROJECT = DEPTHFLOW
 
+_spinner.stop()
+BrokenTorch.manage(DEPTHFLOW.RESOURCES)
+_spinner.start()
+
 # isort: off
+import torch
 from .Modules import *
 from .DepthFlow import *