update

buddy-compiler · Jan 2, 2025 · fe701ef · fe701ef
1 parent c93e481
commit fe701ef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/BuddyStableDiffusion/buddy-stable-diffusion-main.cpp b/examples/BuddyStableDiffusion/buddy-stable-diffusion-main.cpp
@@ -473,7 +473,7 @@ int main() {
     resultVae.getData()[i] = resultVae.getData()[i] * 255;
   }
   intptr_t sizes[4] = {1, 3, 512, 512};
-  Image<float, 4> img(resultVae, sizes);
+  Image<float, 4> img(resultVae.getData(), sizes);
 
   const std::string Imgfilename =
       "../../build/examples/BuddyStableDiffusion/" + image_name + ".bmp";