[MiniCPM3] fix bug

sophgo · Nov 28, 2024 · 8b5b8b4 · 8b5b8b4
1 parent e0e3039
commit 8b5b8b4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/models/MiniCPM3/compile/compile.sh b/models/MiniCPM3/compile/compile.sh
@@ -248,9 +248,9 @@ process_block() {
 for ((i=0; i<$num_layers; i++)); do
     process_block $i &
     models=${models}${outdir}'/block_'$i'.bmodel '$outdir'/block_cache_'$i'.bmodel '
+    sleep 45
 done
 
-sleep 45
 wait  # Wait for all background processes to finish
 
 rm -f *.npz

diff --git a/models/MiniCPM3/requirements.txt b/models/MiniCPM3/requirements.txt
@@ -1,7 +1,7 @@
 transformers_stream_generator
-atamodel_code_generator
+datamodel_code_generator
 accelerate>=0.26.0
 transformers>=4.41.0
 sentencepiece
 einops
-tiktoken
+tiktoken