YongdongWang commited on
Commit
1487628
·
verified ·
1 Parent(s): c3abc84

Update Llama 3.1 8B robot planning space with improvements

Browse files
Files changed (3) hide show
  1. README.md +2 -2
  2. app.py +3 -1
  3. 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.44.0
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.44.0
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=[['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.']],
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.44.0
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