Minor fix

jllllll · Nov 21, 2023 · d2d8e79 · d2d8e79
1 parent f6d1e53
commit d2d8e79
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/.github/workflows/build-wheels-oobabooga.yml b/.github/workflows/build-wheels-oobabooga.yml
@@ -175,10 +175,10 @@ jobs:
           New-Item 'pyproject.toml' -itemType File -value $newpyproject -force
           New-Item 'CMakeLists.txt' -itemType File -value $cmakelists.Replace('llama_cpp','llama_cpp_cuda') -force
           
-          if ($packageVersion -gt [version]'0.2.13' -and $packageVersion -lt [version]'0.2.19')
+          if ($packageVersion -gt [version]'0.2.13')
           {
-            $pyScripts = (dir '.\llama_cpp_cuda\*.py')
-            $pyScripts.fullname.foreach({New-Item $_ -itemType File -value (cat $_ -raw).replace('import llama_cpp.llama','from . import llama') -force})
+            $pyScripts = (Get-ChildItem $(Join-Path '.' 'llama_cpp_cuda' '*.py'))
+            $pyScripts.fullname.foreach({New-Item $_ -itemType File -value (Get-Content $_ -raw).replace('import llama_cpp.llama','from . import llama') -force})
           }
           
       - name: Build Wheel

diff --git a/.github/workflows/build-wheels-rocm-full.yml b/.github/workflows/build-wheels-rocm-full.yml
@@ -170,10 +170,10 @@ jobs:
           New-Item 'pyproject.toml' -itemType File -value $newpyproject -force
           New-Item 'CMakeLists.txt' -itemType File -value $cmakelists.Replace('llama_cpp','llama_cpp_cuda') -force
           
-          if ($packageVersion -gt [version]'0.2.13' -and $packageVersion -lt [version]'0.2.19')
+          if ($packageVersion -gt [version]'0.2.13')
           {
-            $pyScripts = (dir '.\llama_cpp_cuda\*.py')
-            $pyScripts.fullname.foreach({New-Item $_ -itemType File -value (cat $_ -raw).replace('import llama_cpp.llama','from . import llama') -force})
+            $pyScripts = (Get-ChildItem $(Join-Path '.' 'llama_cpp_cuda' '*.py'))
+            $pyScripts.fullname.foreach({New-Item $_ -itemType File -value (Get-Content $_ -raw).replace('import llama_cpp.llama','from . import llama') -force})
           }
         
       - name: Build Wheel