MogensR commited on
Commit
4f3935e
·
1 Parent(s): 1c4e22d

Update core/__init__.py

Browse files
Files changed (1) hide show
  1. core/__init__.py +9 -7
core/__init__.py CHANGED
@@ -11,13 +11,15 @@
11
  OpticalFlowTracker
12
  )
13
 
14
- from .models import (
15
- ModelManager,
16
- SAM2Model,
17
- MatAnyoneModel,
18
- ModelConfig,
19
- ModelCache
20
- )
 
 
21
 
22
  from .quality import (
23
  QualityAnalyzer,
 
11
  OpticalFlowTracker
12
  )
13
 
14
+ # Only import what actually exists in the new models.py
15
+ from .models import ModelManager
16
+
17
+ # Create dummy classes for backward compatibility
18
+ # These are referenced elsewhere but don't exist in the new models.py
19
+ SAM2Model = None
20
+ MatAnyoneModel = None
21
+ ModelConfig = None
22
+ ModelCache = None
23
 
24
  from .quality import (
25
  QualityAnalyzer,