Spaces:
Runtime error
Runtime error
Update Llama 3.1 8B robot planning space with improvements
Browse files- README.md +2 -2
- app.py +3 -1
- requirements.txt +1 -1
README.md
CHANGED
@@ -4,7 +4,7 @@ emoji: 🤖
|
|
4 |
colorFrom: blue
|
5 |
colorTo: green
|
6 |
sdk: gradio
|
7 |
-
sdk_version: 4.
|
8 |
app_file: app.py
|
9 |
pinned: false
|
10 |
license: llama3.1
|
@@ -68,7 +68,7 @@ Convert natural language commands into structured task sequences for constructio
|
|
68 |
- **Architecture**: Llama 3.1 8B + QLoRA adapters
|
69 |
- **Quantization**: 4-bit (NF4) with double quantization
|
70 |
- **Framework**: Transformers + PEFT + BitsAndBytesConfig
|
71 |
-
- **Interface**: Gradio 4.
|
72 |
- **Hardware**: T4-MEDIUM (16GB VRAM)
|
73 |
|
74 |
## ⚡ Performance Notes
|
|
|
4 |
colorFrom: blue
|
5 |
colorTo: green
|
6 |
sdk: gradio
|
7 |
+
sdk_version: 4.32.2
|
8 |
app_file: app.py
|
9 |
pinned: false
|
10 |
license: llama3.1
|
|
|
68 |
- **Architecture**: Llama 3.1 8B + QLoRA adapters
|
69 |
- **Quantization**: 4-bit (NF4) with double quantization
|
70 |
- **Framework**: Transformers + PEFT + BitsAndBytesConfig
|
71 |
+
- **Interface**: Gradio 4.32.2 (stable version)
|
72 |
- **Hardware**: T4-MEDIUM (16GB VRAM)
|
73 |
|
74 |
## ⚡ Performance Notes
|
app.py
CHANGED
@@ -1,3 +1,4 @@
|
|
|
|
1 |
import gradio as gr
|
2 |
import torch
|
3 |
from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
|
@@ -227,7 +228,7 @@ with gr.Blocks(
|
|
227 |
|
228 |
# 示例对话
|
229 |
gr.Examples(
|
230 |
-
examples=[
|
231 |
inputs=msg,
|
232 |
label="💡 Example Robot Commands"
|
233 |
)
|
@@ -254,5 +255,6 @@ if __name__ == "__main__":
|
|
254 |
demo.launch(
|
255 |
server_name="0.0.0.0",
|
256 |
server_port=7860,
|
|
|
257 |
show_error=True
|
258 |
)
|
|
|
1 |
+
|
2 |
import gradio as gr
|
3 |
import torch
|
4 |
from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
|
|
|
228 |
|
229 |
# 示例对话
|
230 |
gr.Examples(
|
231 |
+
examples=['Deploy Excavator 1 to Soil Area 1 for excavation.', 'Send Dump Truck 1 to collect material from Excavator 1, then unload at storage area.', 'Move all robots to avoid Puddle 1 after inspection.', 'Deploy multiple excavators to different soil areas simultaneously.', 'Coordinate dump trucks to transport materials from excavation site to storage.', 'Send robot to inspect rock area, then avoid with all other robots if dangerous.', 'Return all robots to start position after completing tasks.', 'Create a sequence: excavate, load, transport, unload, repeat.'],
|
232 |
inputs=msg,
|
233 |
label="💡 Example Robot Commands"
|
234 |
)
|
|
|
255 |
demo.launch(
|
256 |
server_name="0.0.0.0",
|
257 |
server_port=7860,
|
258 |
+
share=True,
|
259 |
show_error=True
|
260 |
)
|
requirements.txt
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
gradio==4.
|
2 |
transformers==4.44.2
|
3 |
torch==2.1.0
|
4 |
peft==0.7.1
|
|
|
1 |
+
gradio==4.32.2
|
2 |
transformers==4.44.2
|
3 |
torch==2.1.0
|
4 |
peft==0.7.1
|