Update Dockerfile

Dockerfile

@@ -10,7 +10,8 @@ RUN pip install --no-cache-dir \
     accelerate \
     huggingface-hub \
     datasets \
-    tenacity 
+    tenacity \
+    safetensors
 # For retry logic
 
 # Configure cache