diff --git a/builddeepspeed.py b/builddeepspeed.py index 49ea586..286745f 100644 --- a/builddeepspeed.py +++ b/builddeepspeed.py @@ -304,6 +304,9 @@ def run_build_process(self): f.write('set PYTHONIOENCODING=utf-8\n') # Ensure Python uses UTF-8 for its I/O encoding # Call vcvars64.bat explicitly for 64-bit environment f.write(f'call "{vcvars_path}"\n') + #f.write(f'call "{vcvars_path}" x64 -vcvars_ver=14.29\n') + # Fix Git hash + f.write('set DS_BUILD_STRING=+nogit\n') # Set up environment variables f.write(f'set CUDA_PATH=C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v{self.cuda_var.get()}\n') f.write('set CUDA_HOME=%CUDA_PATH%\n') @@ -315,6 +318,7 @@ def run_build_process(self): f.write('set DS_BUILD_FP_QUANTIZER=0\n') f.write('set DS_BUILD_RAGGED_DEVICE_OPS=0\n') f.write('set DS_BUILD_SPARSE_ATTN=0\n') + f.write('set DS_BUILD_TRANSFORMER_INFERENCE=0\n') # Echo environment for debugging f.write('echo ============ Environment Variables ============\n') f.write('echo CUDA_PATH=%CUDA_PATH%\n') @@ -845,4 +849,4 @@ def main(): app.run() if __name__ == "__main__": - main() \ No newline at end of file + main()