chore: change SD_CUBLAS/SD_USE_CUBLAS to SD_CUDA/SD_USE_CUDA

This commit is contained in:
leejet
2024-12-28 13:27:51 +08:00
parent 348a54e34a
commit dcf91f9e0f
6 changed files with 13 additions and 13 deletions

View File

@@ -15,7 +15,7 @@ struct UpscalerGGML {
}
bool load_from_file(const std::string& esrgan_path) {
#ifdef SD_USE_CUBLAS
#ifdef SD_USE_CUDA
LOG_DEBUG("Using CUDA backend");
backend = ggml_backend_cuda_init(0);
#endif