Fix accelerate dependency and device_map compatibility for Spaces

joycaption_app.py

@@ -30,9 +30,9 @@ try:
     
     # Load model with memory-efficient configuration
     model = LlavaForConditionalGeneration.from_pretrained(
-        MODEL_PATH, 
-        torch_dtype="bfloat16", 
-        device_map="auto" if torch.cuda.is_available() else None,
+        MODEL_PATH,
+        torch_dtype="bfloat16",
+        device_map="cpu",  # Force CPU for Spaces compatibility
         load_in_8bit=True,  # Enable 8-bit quantization for memory efficiency
         low_cpu_mem_usage=True,
         trust_remote_code=True

requirements.txt

@@ -3,6 +3,7 @@
 transformers>=4.40.0
 torch>=2.0.0
 torchvision>=0.15.0
+accelerate>=0.20.0  # Required for device_map
 
 # Image processing
 Pillow>=10.0.0