quot;roberta&quot;</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">n_model</span><span class="p">:</span>
            <span class="kn">from</span> <span class="nn">pororo.models.brainbert</span> <span class="kn">import</span> <span class="n">CustomRobertaModel</span>

            <span class="n">model</span> <span class="o">=</span> <span class="p">(</span><span class="n">CustomRobertaModel</span><span