<span class="p">,</span> <span class="nb">min</span><span class="o">=</span><span class="mi">1</span><span class="p">,</span> <span class="nb">max</span><span class="o">=</span><span class="mi">512</span><span class="p">)</span>
<span class="c1"># This particular min, max values for img_id input are recommended by torch dynamo during the export of the model.</span>
<span class="c1"># To see this recommendation, you can try exporting using min=1, max=4096</span>
<span class="n">IMG_ID</span> <span class="o">