Browse Source

Fix small example issues

tags/v0.3.7^2
haixuanTao 1 year ago
parent
commit
c08fe0f3aa
15 changed files with 164 additions and 111 deletions
  1. +1
    -0
      .gitignore
  2. +20
    -20
      examples/piper/piper_left.urdf
  3. +20
    -20
      examples/piper/piper_left_pred.urdf
  4. +20
    -20
      examples/piper/piper_right.urdf
  5. +20
    -20
      examples/piper/piper_right_pred.urdf
  6. +14
    -3
      examples/piper/play_dummy_inference.yml
  7. +1
    -0
      examples/speech-to-text/dataflow.yml
  8. +49
    -0
      examples/translation/dataflow_en_fr.yml
  9. +7
    -5
      examples/vlm/dataflow.yml
  10. +5
    -7
      examples/vlm/dataflow_rerun.yml
  11. +2
    -3
      node-hub/dora-distil-whisper/dora_distil_whisper/main.py
  12. +2
    -1
      node-hub/dora-opus/dora_opus/main.py
  13. +1
    -1
      node-hub/dora-parler/dora_parler/main.py
  14. +1
    -10
      node-hub/dora-qwenvl/dora_qwenvl/main.py
  15. +1
    -1
      node-hub/dora-rerun/pyproject.toml

+ 1
- 0
.gitignore View File

@@ -11,6 +11,7 @@ examples/**/*.txt

# Remove hdf and stl files
*.stl
*.STL
*.hdf5

# Removing images.


examples/piper/assets/piper_left.urdf → examples/piper/piper_left.urdf View File

@@ -23,7 +23,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/base_link.stl" />
filename="base_link.STL" />
</geometry>
<material
name="">
@@ -37,7 +37,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/base_link.stl" />
filename="base_link.STL" />
</geometry>
</collision>
</link>
@@ -63,7 +63,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link1.stl" />
filename="link1.STL" />
</geometry>
<material
name="">
@@ -77,7 +77,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link1.stl" />
filename="link1.STL" />
</geometry>
</collision>
</link>
@@ -121,7 +121,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link2.stl" />
filename="link2.STL" />
</geometry>
<material
name="">
@@ -135,7 +135,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link2.stl" />
filename="link2.STL" />
</geometry>
</collision>
</link>
@@ -179,7 +179,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link3.stl" />
filename="link3.STL" />
</geometry>
<material
name="">
@@ -193,7 +193,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link3.stl" />
filename="link3.STL" />
</geometry>
</collision>
</link>
@@ -237,7 +237,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link4.stl" />
filename="link4.STL" />
</geometry>
<material
name="">
@@ -251,7 +251,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link4.stl" />
filename="link4.STL" />
</geometry>
</collision>
</link>
@@ -295,7 +295,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link5.stl" />
filename="link5.STL" />
</geometry>
<material
name="">
@@ -309,7 +309,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link5.stl" />
filename="link5.STL" />
</geometry>
</collision>
</link>
@@ -353,7 +353,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link6.stl" />
filename="link6.STL" />
</geometry>
<material
name="">
@@ -367,7 +367,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link6.stl" />
filename="link6.STL" />
</geometry>
</collision>
</link>
@@ -412,7 +412,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link7.stl" />
filename="link7.STL" />
</geometry>
<material
name="">
@@ -426,7 +426,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link7.stl" />
filename="link7.STL" />
</geometry>
</collision>
</link>
@@ -471,7 +471,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link8.stl" />
filename="link8.STL" />
</geometry>
<material
name="">
@@ -485,7 +485,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link8.stl" />
filename="link8.STL" />
</geometry>
</collision>
</link>
@@ -529,7 +529,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link8.stl" />
filename="link8.STL" />
</geometry>
<material
name="">
@@ -543,7 +543,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link8.stl" />
filename="link8.STL" />
</geometry>
</collision>
</link>

examples/piper/assets/piper_left_pred.urdf → examples/piper/piper_left_pred.urdf View File

@@ -23,7 +23,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/base_link.stl" />
filename="base_link.STL" />
</geometry>
<material
name="">
@@ -37,7 +37,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/base_link.stl" />
filename="base_link.STL" />
</geometry>
</collision>
</link>
@@ -63,7 +63,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link1.stl" />
filename="link1.STL" />
</geometry>
<material
name="">
@@ -77,7 +77,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link1.stl" />
filename="link1.STL" />
</geometry>
</collision>
</link>
@@ -121,7 +121,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link2.stl" />
filename="link2.STL" />
</geometry>
<material
name="">
@@ -135,7 +135,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link2.stl" />
filename="link2.STL" />
</geometry>
</collision>
</link>
@@ -179,7 +179,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link3.stl" />
filename="link3.STL" />
</geometry>
<material
name="">
@@ -193,7 +193,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link3.stl" />
filename="link3.STL" />
</geometry>
</collision>
</link>
@@ -237,7 +237,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link4.stl" />
filename="link4.STL" />
</geometry>
<material
name="">
@@ -251,7 +251,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link4.stl" />
filename="link4.STL" />
</geometry>
</collision>
</link>
@@ -295,7 +295,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link5.stl" />
filename="link5.STL" />
</geometry>
<material
name="">
@@ -309,7 +309,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link5.stl" />
filename="link5.STL" />
</geometry>
</collision>
</link>
@@ -353,7 +353,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link6.stl" />
filename="link6.STL" />
</geometry>
<material
name="">
@@ -367,7 +367,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link6.stl" />
filename="link6.STL" />
</geometry>
</collision>
</link>
@@ -412,7 +412,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link7.stl" />
filename="link7.STL" />
</geometry>
<material
name="">
@@ -426,7 +426,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link7.stl" />
filename="link7.STL" />
</geometry>
</collision>
</link>
@@ -471,7 +471,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link8.stl" />
filename="link8.STL" />
</geometry>
<material
name="">
@@ -485,7 +485,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link8.stl" />
filename="link8.STL" />
</geometry>
</collision>
</link>
@@ -529,7 +529,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link8.stl" />
filename="link8.STL" />
</geometry>
<material
name="">
@@ -543,7 +543,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link8.stl" />
filename="link8.STL" />
</geometry>
</collision>
</link>

examples/piper/assets/piper_right.urdf → examples/piper/piper_right.urdf View File

@@ -23,7 +23,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/base_link.stl" />
filename="base_link.STL" />
</geometry>
<material
name="">
@@ -37,7 +37,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/base_link.stl" />
filename="base_link.STL" />
</geometry>
</collision>
</link>
@@ -63,7 +63,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link1.stl" />
filename="link1.STL" />
</geometry>
<material
name="">
@@ -77,7 +77,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link1.stl" />
filename="link1.STL" />
</geometry>
</collision>
</link>
@@ -121,7 +121,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link2.stl" />
filename="link2.STL" />
</geometry>
<material
name="">
@@ -135,7 +135,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link2.stl" />
filename="link2.STL" />
</geometry>
</collision>
</link>
@@ -179,7 +179,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link3.stl" />
filename="link3.STL" />
</geometry>
<material
name="">
@@ -193,7 +193,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link3.stl" />
filename="link3.STL" />
</geometry>
</collision>
</link>
@@ -237,7 +237,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link4.stl" />
filename="link4.STL" />
</geometry>
<material
name="">
@@ -251,7 +251,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link4.stl" />
filename="link4.STL" />
</geometry>
</collision>
</link>
@@ -295,7 +295,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link5.stl" />
filename="link5.STL" />
</geometry>
<material
name="">
@@ -309,7 +309,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link5.stl" />
filename="link5.STL" />
</geometry>
</collision>
</link>
@@ -353,7 +353,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link6.stl" />
filename="link6.STL" />
</geometry>
<material
name="">
@@ -367,7 +367,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link6.stl" />
filename="link6.STL" />
</geometry>
</collision>
</link>
@@ -412,7 +412,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link7.stl" />
filename="link7.STL" />
</geometry>
<material
name="">
@@ -426,7 +426,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link7.stl" />
filename="link7.STL" />
</geometry>
</collision>
</link>
@@ -471,7 +471,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link8.stl" />
filename="link8.STL" />
</geometry>
<material
name="">
@@ -485,7 +485,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link8.stl" />
filename="link8.STL" />
</geometry>
</collision>
</link>
@@ -529,7 +529,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link8.stl" />
filename="link8.STL" />
</geometry>
<material
name="">
@@ -543,7 +543,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link8.stl" />
filename="link8.STL" />
</geometry>
</collision>
</link>

examples/piper/assets/piper_right_pred.urdf → examples/piper/piper_right_pred.urdf View File

@@ -23,7 +23,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/base_link.stl" />
filename="base_link.STL" />
</geometry>
<material
name="">
@@ -37,7 +37,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/base_link.stl" />
filename="base_link.STL" />
</geometry>
</collision>
</link>
@@ -63,7 +63,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link1.stl" />
filename="link1.STL" />
</geometry>
<material
name="">
@@ -77,7 +77,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link1.stl" />
filename="link1.STL" />
</geometry>
</collision>
</link>
@@ -121,7 +121,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link2.stl" />
filename="link2.STL" />
</geometry>
<material
name="">
@@ -135,7 +135,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link2.stl" />
filename="link2.STL" />
</geometry>
</collision>
</link>
@@ -179,7 +179,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link3.stl" />
filename="link3.STL" />
</geometry>
<material
name="">
@@ -193,7 +193,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link3.stl" />
filename="link3.STL" />
</geometry>
</collision>
</link>
@@ -237,7 +237,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link4.stl" />
filename="link4.STL" />
</geometry>
<material
name="">
@@ -251,7 +251,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link4.stl" />
filename="link4.STL" />
</geometry>
</collision>
</link>
@@ -295,7 +295,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link5.stl" />
filename="link5.STL" />
</geometry>
<material
name="">
@@ -309,7 +309,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link5.stl" />
filename="link5.STL" />
</geometry>
</collision>
</link>
@@ -353,7 +353,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link6.stl" />
filename="link6.STL" />
</geometry>
<material
name="">
@@ -367,7 +367,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link6.stl" />
filename="link6.STL" />
</geometry>
</collision>
</link>
@@ -412,7 +412,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link7.stl" />
filename="link7.STL" />
</geometry>
<material
name="">
@@ -426,7 +426,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link7.stl" />
filename="link7.STL" />
</geometry>
</collision>
</link>
@@ -471,7 +471,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link8.stl" />
filename="link8.STL" />
</geometry>
<material
name="">
@@ -485,7 +485,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link8.stl" />
filename="link8.STL" />
</geometry>
</collision>
</link>
@@ -529,7 +529,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link8.stl" />
filename="link8.STL" />
</geometry>
<material
name="">
@@ -543,7 +543,7 @@
rpy="0 0 0" />
<geometry>
<mesh
filename="assets/link8.stl" />
filename="link8.STL" />
</geometry>
</collision>
</link>

+ 14
- 3
examples/piper/play_dummy_inference.yml View File

@@ -10,13 +10,24 @@ nodes:
CAN_BUS: can_left

- id: rerun
build: wget
build: |
proxychains wget -nc https://raw.githubusercontent.com/agilexrobotics/Piper_ros/4f22c61f96b8fb3ef3f937b99b63edb697caadf0/src/piper_description/meshes/base_link.STL
proxychains wget -nc https://raw.githubusercontent.com/agilexrobotics/Piper_ros/4f22c61f96b8fb3ef3f937b99b63edb697caadf0/src/piper_description/meshes/link1.STL
proxychains wget -nc https://raw.githubusercontent.com/agilexrobotics/Piper_ros/4f22c61f96b8fb3ef3f937b99b63edb697caadf0/src/piper_description/meshes/link2.STL
proxychains wget -nc https://raw.githubusercontent.com/agilexrobotics/Piper_ros/4f22c61f96b8fb3ef3f937b99b63edb697caadf0/src/piper_description/meshes/link3.STL
proxychains wget -nc https://raw.githubusercontent.com/agilexrobotics/Piper_ros/4f22c61f96b8fb3ef3f937b99b63edb697caadf0/src/piper_description/meshes/link4.STL
proxychains wget -nc https://raw.githubusercontent.com/agilexrobotics/Piper_ros/4f22c61f96b8fb3ef3f937b99b63edb697caadf0/src/piper_description/meshes/link5.STL
proxychains wget -nc https://raw.githubusercontent.com/agilexrobotics/Piper_ros/4f22c61f96b8fb3ef3f937b99b63edb697caadf0/src/piper_description/meshes/link6.STL
proxychains wget -nc https://raw.githubusercontent.com/agilexrobotics/Piper_ros/4f22c61f96b8fb3ef3f937b99b63edb697caadf0/src/piper_description/meshes/link7.STL
proxychains wget -nc https://raw.githubusercontent.com/agilexrobotics/Piper_ros/4f22c61f96b8fb3ef3f937b99b63edb697caadf0/src/piper_description/meshes/link8.STL
pip install -e ../../node-hub/dora-rerun
pip install git+https://github.com/rerun-io/rerun-loader-python-example-urdf.git
path: dora-rerun
inputs:
jointstate_piper_left: piper/jointstate_left
jointstate_piper_right: piper/jointstate_right
env:
piper_left_urdf: assets/piper_left.urdf # Make sure to download assets/meshes from https://github.com/agilexrobotics/Piper_ros/tree/4f22c61f96b8fb3ef3f937b99b63edb697caadf0/src/piper_description/meshes and put them in the assets folder
piper_right_urdf: assets/piper_right.urdf # Make sure to download assets/meshes from https://github.com/agilexrobotics/Piper_ros/tree/4f22c61f96b8fb3ef3f937b99b63edb697caadf0/src/piper_description/meshes and put them in the assets folder
piper_left_urdf: piper_left.urdf # Make sure to download meshes from https://github.com/agilexrobotics/Piper_ros/tree/4f22c61f96b8fb3ef3f937b99b63edb697caadf0/src/piper_description/meshes and put them in the assets folder
piper_right_urdf: piper_right.urdf # Make sure to download meshes from https://github.com/agilexrobotics/Piper_ros/tree/4f22c61f96b8fb3ef3f937b99b63edb697caadf0/src/piper_description/meshes and put them in the assets folder
piper_left_transform: 0 0.2 0
piper_right_transform: 0 -0.2 0

+ 1
- 0
examples/speech-to-text/dataflow.yml View File

@@ -22,6 +22,7 @@ nodes:
- text
env:
TARGET_LANGUAGE: english
# For China
# USE_MODELSCOPE_HUB: true

- id: dora-rerun


+ 49
- 0
examples/translation/dataflow_en_fr.yml View File

@@ -0,0 +1,49 @@
nodes:
- id: dora-microphone
build: pip install -e ../../node-hub/dora-microphone
path: dora-microphone
outputs:
- audio

- id: dora-vad
build: pip install -e ../../node-hub/dora-vad
path: dora-vad
inputs:
audio: dora-microphone/audio
outputs:
- audio

- id: dora-distil-whisper
build: pip install -e ../../node-hub/dora-distil-whisper
path: dora-distil-whisper
inputs:
input: dora-vad/audio
outputs:
- text
env:
TARGET_LANGUAGE: english
# For China
# HF_ENDPOINT: https://hf-mirror.com

- id: dora-opus
build: pip install -e ../../node-hub/dora-opus
path: dora-opus
inputs:
text: dora-distil-whisper/text
outputs:
- text
env:
SOURCE_LANGUAGE: en
TARGET_LANGUAGE: fr
# For China
# HF_ENDPOINT: https://hf-mirror.com

- id: dora-rerun
build: cargo build -p dora-rerun --release
path: dora-rerun
inputs:
translated_text: dora-opus/text
original_text: dora-distil-whisper/text
env:
IMAGE_WIDTH: 640
IMAGE_HEIGHT: 480

+ 7
- 5
examples/vlm/dataflow.yml View File

@@ -1,9 +1,9 @@
nodes:
- id: camera
build: pip install ../../node-hub/opencv-video-capture
build: pip install -e ../../node-hub/opencv-video-capture
path: opencv-video-capture
inputs:
tick: dora/timer/millis/20
tick: dora/timer/millis/50
outputs:
- image
env:
@@ -18,15 +18,17 @@ nodes:
image:
source: camera/image
queue_size: 1
tick: dora/timer/millis/100
tick: dora/timer/millis/400
outputs:
- text
- tick
env:
DEFAULT_QUESTION: Describe the image.
DEFAULT_QUESTION: Describe the image in a very short sentence.
# For China
# USE_MODELSCOPE_HUB: true

- id: plot
build: pip install ../../node-hub/opencv-plot
build: pip install -e ../../node-hub/opencv-plot
path: opencv-plot
inputs:
image:


+ 5
- 7
examples/vlm/dataflow_rerun.yml View File

@@ -3,7 +3,7 @@ nodes:
build: pip install -e ../../node-hub/opencv-video-capture
path: opencv-video-capture
inputs:
tick: dora/timer/millis/20
tick: dora/timer/millis/50
outputs:
- image
env:
@@ -18,12 +18,13 @@ nodes:
image:
source: camera/image
queue_size: 1
tick: dora/timer/millis/200
tick: dora/timer/millis/400
outputs:
- text
- tick
env:
DEFAULT_QUESTION: Describe the image.
DEFAULT_QUESTION: Describe the image in a very short sentence.
# USE_MODELSCOPE_HUB: true

- id: plot
build: cargo build -p dora-rerun --release
@@ -37,7 +38,4 @@ nodes:
IMAGE_WIDTH: 640
IMAGE_HEIGHT: 480
README: |
# Visualization of your webcam

- You can increase or decrease the frequency within the dataflow with: `tick: dora/timer/millis/20` in the webcam node.
- You can change the model prompt within the dataflow with: `DEFAULT_QUESTION: Describe the image.`
# Visualization of QwenVL2

+ 2
- 3
node-hub/dora-distil-whisper/dora_distil_whisper/main.py View File

@@ -7,12 +7,12 @@ from pathlib import Path

DEFAULT_PATH = "openai/whisper-large-v3-turbo"
TARGET_LANGUAGE = os.getenv("TARGET_LANGUAGE", "chinese")
TRANSLATE = bool(os.getenv("TRANSLATE", "False"))
TRANSLATE = bool(os.getenv("TRANSLATE", "False") in ["True", "true"])


MODEL_NAME_OR_PATH = os.getenv("MODEL_NAME_OR_PATH", DEFAULT_PATH)

if bool(os.getenv("USE_MODELSCOPE_HUB")) is True:
if bool(os.getenv("USE_MODELSCOPE_HUB") in ["True", "true"]):
from modelscope import snapshot_download

if not Path(MODEL_NAME_OR_PATH).exists():
@@ -107,5 +107,4 @@ def main():
if result["text"] in BAD_SENTENCES:
continue
text = cut_repetition(result["text"])
print(text, flush=True)
node.send_output("text", pa.array([text]), {"language": TARGET_LANGUAGE})

+ 2
- 1
node-hub/dora-opus/dora_opus/main.py View File

@@ -5,6 +5,7 @@ import pyarrow as pa
import numpy as np
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM


from_code = os.getenv("SOURCE_LANGUAGE", "zh")
to_code = os.getenv("TARGET_LANGUAGE", "en")
DEFAULT_PATH = f"Helsinki-NLP/opus-mt-{from_code}-{to_code}"
@@ -12,7 +13,7 @@ DEFAULT_PATH = f"Helsinki-NLP/opus-mt-{from_code}-{to_code}"

MODEL_NAME_OR_PATH = os.getenv("MODEL_NAME_OR_PATH", DEFAULT_PATH)

if bool(os.getenv("USE_MODELSCOPE_HUB")) is True:
if bool(os.getenv("USE_MODELSCOPE_HUB") in ["True", "true"]):
from modelscope import snapshot_download

if not Path(MODEL_NAME_OR_PATH).exists():


+ 1
- 1
node-hub/dora-parler/dora_parler/main.py View File

@@ -24,7 +24,7 @@ DEFAULT_PATH = "ylacombe/parler-tts-mini-jenny-30H"

MODEL_NAME_OR_PATH = os.getenv("MODEL_NAME_OR_PATH", DEFAULT_PATH)

if bool(os.getenv("USE_MODELSCOPE_HUB")) is True:
if bool(os.getenv("USE_MODELSCOPE_HUB") in ["True", "true"]):
from modelscope import snapshot_download

if not Path(MODEL_NAME_OR_PATH).exists():


+ 1
- 10
node-hub/dora-qwenvl/dora_qwenvl/main.py View File

@@ -12,7 +12,7 @@ DEFAULT_PATH = "Qwen/Qwen2-VL-2B-Instruct"

MODEL_NAME_OR_PATH = os.getenv("MODEL_NAME_OR_PATH", DEFAULT_PATH)

if bool(os.getenv("USE_MODELSCOPE_HUB")) is True:
if bool(os.getenv("USE_MODELSCOPE_HUB") in ["True", "true"]):
from modelscope import snapshot_download

if not Path(MODEL_NAME_OR_PATH).exists():
@@ -111,15 +111,6 @@ def main():

if event_type == "INPUT":

# pylint: disable=fixme
# TODO: Remove this after https://github.com/dora-rs/dora/pull/652
while True:
next_event = node.next(timeout=0.001)
if next_event is not None and next_event["type"] == "INPUT":
event = next_event
else:
break

event_id = event["id"]

if "image" in event_id:


+ 1
- 1
node-hub/dora-rerun/pyproject.toml View File

@@ -12,4 +12,4 @@ dependencies = [
scripts = { "dora-rerun" = "dora_rerun:py_main" }

[tool.maturin]
features = ["pyo3/extension-module"]
features = ["python", "pyo3/extension-module"]

Loading…
Cancel
Save