Skip local LoRA download on HF Spaces, clean up training uploads

LoRA is saved to RunPod network volume (ready for generation).
Local download skipped on HF to avoid filling 1GB storage limit.
Training upload images cleaned up after upload to RunPod.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

src/content_engine/services/runpod_trainer.py

@@ -654,14 +654,17 @@ resolution = [{resolution}, {resolution}]
             self._ssh_exec(ssh, f"cp {remote_output} /runpod-volume/loras/{name}.safetensors")
             job._log(f"LoRA saved to volume: /runpod-volume/loras/{name}.safetensors")
 
-            # Then download locally
-            job._log("Downloading LoRA to local machine...")
-            LORA_OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
-            local_path = LORA_OUTPUT_DIR / f"{name}.safetensors"
-            sftp.get(remote_output, str(local_path))
-
-            job.output_path = str(local_path)
-            job._log(f"LoRA saved locally to {local_path}")
+            # Download locally (skip on HF Spaces — limited storage)
+            if IS_HF_SPACES:
+                job.output_path = f"/runpod-volume/loras/{name}.safetensors"
+                job._log("LoRA saved on RunPod volume (ready for generation)")
+            else:
+                job._log("Downloading LoRA to local machine...")
+                LORA_OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
+                local_path = LORA_OUTPUT_DIR / f"{name}.safetensors"
+                sftp.get(remote_output, str(local_path))
+                job.output_path = str(local_path)
+                job._log(f"LoRA saved locally to {local_path}")
 
             # Done!
             job.status = "completed"
@@ -689,6 +692,13 @@ resolution = [{resolution}, {resolution}]
                 except Exception:
                     pass
 
+            # Clean up local training images (saves HF Spaces storage)
+            if image_paths:
+                import shutil
+                first_image_dir = Path(image_paths[0]).parent
+                if first_image_dir.exists() and "training_uploads" in str(first_image_dir):
+                    shutil.rmtree(first_image_dir, ignore_errors=True)
+
             if job.pod_id:
                 try:
                     job._log("Terminating RunPod...")