Upload folder using huggingface_hub

requirements.txt

@@ -1,3 +1,4 @@
+torch>=2.0.0
 accelerate>=0.27.0
 bitsandbytes>=0.41.0
 datasets>=2.15.0
@@ -18,7 +19,6 @@ requests>=2.31.0
 safetensors>=0.4.1
 sentencepiece>=0.1.99
 tensorboard>=2.15.0
-torch>=2.0.0
 tqdm>=4.65.0
 transformers>=4.36.0
 typing-extensions>=4.8.0

update_space.py

@@ -121,12 +121,17 @@ def update_requirements():
     # Add new requirements
     updated_requirements = existing_requirements.union(required_packages)
     
-    # Write updated requirements
+    # Write updated requirements with torch first
     with open(req_path, 'w') as f:
-        for req in sorted(updated_requirements):
+        # Ensure torch is first
+        torch_req = next((req for req in updated_requirements if req.startswith("torch")), "torch>=2.0.0")
+        f.write(f"{torch_req}\n")
+        
+        # Write remaining requirements
+        for req in sorted(r for r in updated_requirements if not r.startswith("torch")):
             f.write(f"{req}\n")
     
-    logger.info("Updated requirements.txt with necessary packages")
+    logger.info("Updated requirements.txt with necessary packages (torch listed first)")
 
 def create_space(username, space_name):
     """Create or get a Hugging Face Space."""