Address changes from review

Author: a-szymanska

packages/react-native-executorch/common/rnexecutorch/models/text_to_image/Encoder.cpp

@@ -20,6 +20,7 @@ std::vector<float> Encoder::generate(std::string input) {
   std::shared_ptr<OwningArrayBuffer> embeddingsUncond =
       encoder->generate(std::string(constants::kBosToken));
 
+  assert(embeddingsText->size() == embeddingsUncond->size());
   size_t embeddingsSize = embeddingsText->size() / sizeof(float);
   auto *embeddingsTextPtr = reinterpret_cast<float *>(embeddingsText->data());
   auto *embeddingsUncondPtr =

packages/react-native-executorch/common/rnexecutorch/models/text_to_image/TextToImage.cpp

@@ -97,7 +97,7 @@ TextToImage::postprocess(const std::vector<float> &output) const {
     outputRgb[i * 4 + 3] = 255;
   }
 
-  // Replace with a function when #584 implemented
+  // TODO: Replace with a function when #584 implemented
   auto createBuffer = [](const auto &data, size_t size) {
     auto buffer = std::make_shared<OwningArrayBuffer>(size);
     std::memcpy(buffer->data(), data, size);
@@ -106,7 +106,7 @@ TextToImage::postprocess(const std::vector<float> &output) const {
   return createBuffer(outputRgb.data(), outputRgb.size() * sizeof(float));
 }
 
-void TextToImage::interrupt() { interrupted = true; }
+void TextToImage::interrupt() noexcept { interrupted = true; }
 
 size_t TextToImage::getMemoryLowerBound() const noexcept {
   return encoder->getMemoryLowerBound() + unet->getMemoryLowerBound() +

packages/react-native-executorch/common/rnexecutorch/models/text_to_image/TextToImage.h

@@ -32,7 +32,7 @@ class TextToImage final {
   std::shared_ptr<OwningArrayBuffer>
   generate(std::string input, size_t numInferenceSteps,
            std::shared_ptr<jsi::Function> callback);
-  void interrupt();
+  void interrupt() noexcept;
   size_t getMemoryLowerBound() const noexcept;
   void unload() noexcept;
 

packages/react-native-executorch/src/constants/modelUrls.ts

@@ -424,7 +424,7 @@ export const CLIP_VIT_BASE_PATCH32_TEXT = {
 };
 
 // Image generation
-export const BK_SDM_TINY_VPRED = {
+export const BK_SDM_TINY_VPRED_512 = {
   schedulerSource:
     'https://huggingface.co/aszymanska/bk-sdm-tiny-vpred/resolve/main/scheduler/scheduler_config.json',
   tokenizerSource: