Update docs from 5d3dfcd

microsoft · Dec 11, 2023 · 920905f · 920905f
1 parent d0e2c0d
commit 920905f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/_modules/olive/systems/local.html b/_modules/olive/systems/local.html
@@ -168,7 +168,7 @@ <h1>Source code for olive.systems.local</h1><div class="highlight"><pre>
         <span class="bp">self</span><span class="p">,</span> <span class="n">model_config</span><span class="p">:</span> <span class="n">ModelConfig</span><span class="p">,</span> <span class="n">data_root</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">metrics</span><span class="p">:</span> <span class="n">List</span><span class="p">[</span><span class="s2">&quot;Metric&quot;</span><span class="p">],</span> <span class="n">accelerator</span><span class="p">:</span> <span class="n">AcceleratorSpec</span>
     <span class="p">)</span> <span class="o">-&gt;</span> <span class="s2">&quot;MetricResult&quot;</span><span class="p">:</span>
 <span class="w">        </span><span class="sd">&quot;&quot;&quot;Evaluate the model.&quot;&quot;&quot;</span>
-        <span class="k">if</span> <span class="n">model_config</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">lower</span><span class="p">()</span> <span class="o">==</span> <span class="s2">&quot;CompositeOnnxModel&quot;</span><span class="o">.</span><span class="n">lower</span><span class="p">():</span>
+        <span class="k">if</span> <span class="n">model_config</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">lower</span><span class="p">()</span> <span class="ow">in</span> <span class="p">(</span><span class="s2">&quot;compositemodel&quot;</span><span class="p">,</span> <span class="s2">&quot;compositepytorchmodel&quot;</span><span class="p">):</span>
             <span class="k">raise</span> <span class="ne">NotImplementedError</span>
 
         <span class="n">device</span> <span class="o">=</span> <span class="n">accelerator</span><span class="o">.</span><span class="n">accelerator_type</span> <span class="k">if</span> <span class="n">accelerator</span> <span class="k">else</span> <span class="n">Device</span><span class="o">.</span><span class="n">CPU</span>