Koan-Sin Tan
Koan-Sin Tan
> I think there is an answer here that it is working: [keras-team/keras#19108](https://github.com/keras-team/keras/issues/19108) It is based on this comment: [keras-team/keras#19108 (comment)](https://github.com/keras-team/keras/issues/19108#issuecomment-1913421572) > > So my code now is: > >...
> @freedomtan can you share your working colab here nope, because I tested it with a simple script on my local machine; didn't try to deal with memory issues in...
@charbull I was able to convert `text_encoder`, `diffusion_model`, and `encoder` to tflite. There are some issues. 1. you have to specify batch size some where, otherwise, you got error messages...
> Hi @freedomtan how do you set the batch size ? the arguments are: > > ``` > img_height: Height of the images to generate, in pixel. Note that only...
> I wonder why the 2GB limitation will be applied here, since weights is not part of the proto? Yup, that's not trivial. In TensorFlow 2.10 and before (I didn't...
> > I think in master is https://github.com/tensorflow/tensorflow/blob/master/tensorflow/compiler/mlir/lite/flatbuffer_export.cc#L2152-L2156 > > you mean the 2GB limit is still true across 2.10 and master, correct? Nope, there are two 2 GiB limitations,...
@charbull, my 2 cents: DO NOT use `tf.lite.OpsSet.SELECT_TF_OPS`, which is to allow TF ops. Since your goal is to run on EdgeTPU, TF ops are unlikely to work on it....
@charbull What you saved in saved_model is fp32 model. If you check input tensors of the diffusion_model with `saved_model_cli`, you can see that the expected data types of the 3...
@gbaned from `git blame tensorflow/lite/tools/evaluation/stages/image_preprocessing_stage.cc`, the file was mainly touched by @srjoglekar246 and @thaink
Hi @gbaned, is there anything I can do to move forward?