ss="p">,</span> <span class="s2">&quot;tx&quot;</span><span class="p">)]</span>
    <span class="p">)</span>
    <span class="n">lit_segger</span> <span class="o">=</span> <span class="n">LitSegger</span><span class="p">(</span>
        <span class="n">init_emb</span><span class="o">=</span><span class="n">args</span><span class="o">.</span><span class="n">init_emb</span><span class="p">,</span>
        <span class="n">hidden_channels</span><span class="o">=</span><span class="n">args</span><span class="o">