Upload 3 files

ai_http.py

@@ -55,7 +55,17 @@ class AIAccelerator:
         if not self.vram:
             return True  # No VRAM restrictions
             
-        if self._vram_allocated + size_bytes > self.vram.total_size:
+        # Check vram_state for unlimited allocation
+        if hasattr(self.vram, 'vram_state') and self.vram.vram_state.get('is_unlimited', False):
+            self._vram_allocated += size_bytes
+            return True
+            
+        # If there's a specific size limit in vram_state
+        total_size = float('inf')
+        if hasattr(self.vram, 'vram_state'):
+            total_size = self.vram.vram_state.get('total_size', float('inf'))
+            
+        if self._vram_allocated + size_bytes > total_size:
             return False
             
         self._vram_allocated += size_bytes

test_ai_integration_http.py

@@ -122,7 +122,7 @@ def test_ai_integration_http():
             components['chips'].append(chip_for_loading)
             
             # Initialize VRAM with local storage (unlimited)
-            vram = VirtualVRAM(storage=storage)
+            vram = VirtualVRAM(size_gb=None, storage=storage)  # Explicitly set unlimited
             components['vram'] = vram
             
             # Set up AI accelerator

virtual_vram.py

@@ -15,14 +15,24 @@ class VirtualVRAM:
         
         # Initialize VRAM state with unlimited capacity
         self.vram_state = {
-            "total_size": float('inf'),  # Unlimited size
+            "total_size": size_gb * 1024 * 1024 * 1024 if size_gb is not None else float('inf'),  # Convert GB to bytes or use unlimited
             "allocated": 0,
             "blocks": {},
             "memory_map": {},
-            "is_unlimited": True
+            "is_unlimited": size_gb is None
         }
         self.store_vram_state()
         
+    @property
+    def total_size(self) -> int:
+        """Get total VRAM size in bytes"""
+        return self.vram_state["total_size"]
+        
+    @property
+    def available_size(self) -> int:
+        """Get available VRAM size in bytes"""
+        return self.vram_state["total_size"] - self.vram_state["allocated"] if not self.vram_state["is_unlimited"] else float('inf')
+        
     def store_vram_state(self, max_retries=3):
         """Store VRAM state in WebSocket storage with retry logic"""
         for attempt in range(max_retries):