filter_layers</span><span class="o">=</span><span class="n">LAYERS</span><span class="p">,</span>
<span class="lineno">69</span>                                 <span class="n">dtype</span><span class="o">=</span><span class="n">torch</span><span class="o">.</span><span class="n">float16</span><span class="p">,</span>
<span class="lineno">70</span>                                 <span class="n">device</span><span class="o">=</span><span class="n">device</span><span class="p">,</span>
<span class="lineno">7