README.md

metadata

title: Robot Task Planning - Llama 3.1 8B
emoji: 🤖
colorFrom: blue
colorTo: green
sdk: gradio
sdk_version: 4.32.2
app_file: app.py
pinned: false
license: llama3.1
hardware: t4-medium

🤖 Robot Task Planning - Llama 3.1 8B

This Space demonstrates a fine-tuned version of Meta's Llama 3.1 8B model specialized for robot task planning using QLoRA (4-bit quantization + LoRA) technique.

🎯 Purpose

Convert natural language commands into structured task sequences for construction robots including:

• Excavators - Digging, loading, positioning
• Dump Trucks - Material transport, loading, unloading
• Multi-robot Coordination - Complex task dependencies

🔗 Model

Fine-tuned Model: YongdongWang/llama-3.1-8b-dart-qlora

Base Model: meta-llama/Llama-3.1-8B

✨ Features

• 🎮 Interactive Chat Interface - Real-time robot command processing
• ⚙️ Configurable Generation - Adjust temperature, top-p, max tokens
• 📝 Example Commands - Pre-built scenarios to get started
• 🚀 Optimized Performance - 4-bit quantization for efficient inference
• 📊 Structured Output - JSON-formatted task sequences

🚀 Usage

1. Input: Natural language robot commands

   "Deploy Excavator 1 to Soil Area 1 for excavation"
   

2. Output: Structured task sequences

   {
     "tasks": [
       {
         "robot": "Excavator_1",
         "action": "move_to",
         "target": "Soil_Area_1",
         "duration": 30
       },
       {
         "robot": "Excavator_1", 
         "action": "excavate",
         "target": "Soil_Area_1",
         "duration": 120
       }
     ]
   }
   

🛠️ Technical Details

• Architecture: Llama 3.1 8B + QLoRA adapters
• Quantization: 4-bit (NF4) with double quantization
• Framework: Transformers + PEFT + BitsAndBytesConfig
• Interface: Gradio 4.32.2 (stable version)
• Hardware: T4-MEDIUM (16GB VRAM)

⚡ Performance Notes

• First Load: 3-5 minutes (model downloading + loading)
• Subsequent Generations: ~2-10 seconds per response
• Memory Usage: ~8GB VRAM with 4-bit quantization
• Context Length: Up to 2048 tokens

📚 Example Commands

Try these robot commands:

• "Deploy Excavator 1 to Soil Area 1 for excavation"
• "Send Dump Truck 1 to collect material, then unload at storage"
• "Coordinate multiple excavators across different areas"
• "Create evacuation sequence for all robots from dangerous zone"

🔬 Research Applications

This model demonstrates:

• Natural Language → Robot Planning translation
• Multi-agent Task Coordination
• Efficient LLM Fine-tuning with QLoRA
• Real-time Robot Command Processing

📄 License

This project uses Meta's Llama 3.1 license. Please review the license terms before use.

🤝 Contributing

For issues, improvements, or questions about the model, please visit the model repository.