WildnerveAI commited on
Commit
c8e5b57
·
verified ·
1 Parent(s): 6ffc9f3

Upload adapter_layer.py

Browse files
Files changed (1) hide show
  1. adapter_layer.py +19 -2
adapter_layer.py CHANGED
@@ -13,9 +13,26 @@ os.environ["LOW_MEMORY_MODE"] = "1"
13
  # Log versions and fail fast if missing
14
  logger = logging.getLogger(__name__)
15
  logger.info(f"Using pydantic v{pydantic.__version__}")
16
- logger.info(f"Using codecarbon v{codecarbon.__version__}")
17
 
18
- print(f"Successfully using installed dependencies - pydantic: {pydantic.__version__}, codecarbon: {codecarbon.__version__}")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
19
 
20
  # MEMORY OPTIMIZATION: Show current memory usage
21
  def log_memory_usage():
 
13
  # Log versions and fail fast if missing
14
  logger = logging.getLogger(__name__)
15
  logger.info(f"Using pydantic v{pydantic.__version__}")
 
16
 
17
+ # Add proper codecarbon import handling
18
+ try:
19
+ import codecarbon
20
+ codecarbon_available = True
21
+ logger.info(f"Using codecarbon v{codecarbon.__version__}")
22
+ except ImportError:
23
+ codecarbon_available = False
24
+ logger.warning("codecarbon is not available - carbon tracking disabled")
25
+ # Create dummy class for compatibility
26
+ class DummyEmissionsTracker:
27
+ def __init__(self, *args, **kwargs): pass
28
+ def start(self): return self
29
+ def stop(self): return 0.0
30
+
31
+ class codecarbon:
32
+ __version__ = "unavailable"
33
+ EmissionsTracker = DummyEmissionsTracker
34
+
35
+ print(f"Successfully using installed dependencies - pydantic: {pydantic.__version__}, codecarbon: {'available' if codecarbon_available else 'unavailable'}")
36
 
37
  # MEMORY OPTIMIZATION: Show current memory usage
38
  def log_memory_usage():