Explorar o código

clip : use ggml_backend_buffer_is_host (#4205)

Georgi Gerganov %!s(int64=2) %!d(string=hai) anos
pai
achega
0235b9b571
Modificáronse 1 ficheiros con 1 adicións e 5 borrados
  1. 1 5
      examples/llava/clip.cpp

+ 1 - 5
examples/llava/clip.cpp

@@ -598,11 +598,7 @@ struct clip_ctx * clip_model_load(const char * fname, const int verbosity = 1) {
                 return nullptr;
                 return nullptr;
             }
             }
             int num_bytes = ggml_nbytes(cur);
             int num_bytes = ggml_nbytes(cur);
-            if (ggml_backend_is_cpu(new_clip->backend)
-#ifdef GGML_USE_METAL
-            || ggml_backend_is_metal(new_clip->backend)
-#endif
-            ) {
+            if (ggml_backend_buffer_is_host(new_clip->params_buffer)) {
                 // for the CPU and Metal backend, we can read directly into the tensor
                 // for the CPU and Metal backend, we can read directly into the tensor
                 fin.read(reinterpret_cast<char *>(cur->data), num_bytes);
                 fin.read(reinterpret_cast<char *>(cur->data), num_bytes);
             } else {
             } else {