Add fix for #21

jllllll · Nov 17, 2023 · c791889 · c791889
1 parent cefbdb0
commit c791889
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 2 deletions.
diff --git a/.github/workflows/build-wheels-batch-oobabooga.yml b/.github/workflows/build-wheels-batch-oobabooga.yml
@@ -5,7 +5,7 @@ on:
     inputs:
       versions:
         description: 'Comma-seperated version tags of llama-cpp-python to build'
-        default: 'v0.1.77,v0.1.76'
+        default: 'v0.2.18,v0.2.17,v0.2.16,v0.2.15,v0.2.14'
         required: true
         type: string
 

diff --git a/.github/workflows/build-wheels-batch-rocm.yml b/.github/workflows/build-wheels-batch-rocm.yml
@@ -5,7 +5,7 @@ on:
     inputs:
       versions:
         description: 'Comma-seperated version tags of llama-cpp-python to build'
-        default: 'v0.2.11,v0.2.10,v0.2.9,v0.2.8'
+        default: 'v0.2.18,v0.2.17,v0.2.16,v0.2.15,v0.2.14'
         required: true
         type: string
       config:

diff --git a/.github/workflows/build-wheels-oobabooga.yml b/.github/workflows/build-wheels-oobabooga.yml
@@ -175,6 +175,12 @@ 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')
+          {
+            $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})
+          }
+          
       - name: Build Wheel
         run: |
           $packageVersion = [version]$env:PCKGVER.TrimStart('v')

diff --git a/.github/workflows/build-wheels-rocm-full.yml b/.github/workflows/build-wheels-rocm-full.yml
@@ -169,6 +169,12 @@ jobs:
           Copy-Item 'llama_cpp' 'llama_cpp_cuda' -recurse
           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')
+          {
+            $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})
+          }
         
       - name: Build Wheel
         run: |