fix wheel build
This commit is contained in:
		
							parent
							
								
									8699de6106
								
							
						
					
					
						commit
						5157b341a0
					
				
							
								
								
									
										13
									
								
								setup.py
								
								
								
								
							
							
						
						
									
										13
									
								
								setup.py
								
								
								
								
							|  | @ -1,4 +1,5 @@ | |||
| from distutils.core import setup | ||||
| from distutils.extension import Extension | ||||
| from Cython.Build import cythonize | ||||
| import numpy, os, sys | ||||
| 
 | ||||
|  | @ -10,6 +11,17 @@ else: | |||
|     os.environ['CFLAGS']   = '-mavx -mavx2 -mfma -mf16c -O3 -std=c11' | ||||
|     os.environ['CXXFLAGS'] = '-mavx -mavx2 -mfma -mf16c -O3 -std=c++11' | ||||
| 
 | ||||
| ext_modules = [ | ||||
|     Extension( | ||||
|         name="whispercpp", | ||||
|         sources=["whispercpp.pyx", "whisper.cpp/whisper.cpp"], | ||||
|         language="c++", | ||||
|         extra_compile_args=["-std=c++11"], | ||||
|    ) | ||||
| ] | ||||
| ext_modules = cythonize(ext_modules) | ||||
| 
 | ||||
| whisper_clib = ('whisper_clib', {'sources': ['whisper.cpp/ggml.c']}) | ||||
| 
 | ||||
| setup( | ||||
|     name='whispercpp', | ||||
|  | @ -17,6 +29,7 @@ setup( | |||
|     description='Python bindings for whisper.cpp', | ||||
|     author='Luke Southam', | ||||
|     author_email='luke@devthe.com', | ||||
|     libraries=[whisper_clib], | ||||
|     ext_modules = cythonize("whispercpp.pyx"), | ||||
|     include_dirs = ['./whisper.cpp/', numpy.get_include()], | ||||
|     install_requires=[ | ||||
|  |  | |||
|  | @ -1,7 +1,5 @@ | |||
| #!python | ||||
| # cython: language_level=3 | ||||
| # distutils: language = c++ | ||||
| # distutils: sources= ./whisper.cpp/whisper.cpp ./whisper.cpp/ggml.c | ||||
| 
 | ||||
| import ffmpeg | ||||
| import numpy as np | ||||
|  |  | |||
		Loading…
	
		Reference in New Issue