deploy: 8973795

Author: liugangcode

_modules/index.html

@@ -264,28 +264,14 @@ <h1>All modules for which code is available</h1>
 <li><a href="torch_molecule/base/encoder.html">torch_molecule.base.encoder</a></li>
 <li><a href="torch_molecule/base/generator.html">torch_molecule.base.generator</a></li>
 <li><a href="torch_molecule/base/predictor.html">torch_molecule.base.predictor</a></li>
-<li><a href="torch_molecule/encoder/attrmask/modeling_attrmask.html">torch_molecule.encoder.attrmask.modeling_attrmask</a></li>
-<li><a href="torch_molecule/encoder/contextpred/modeling_contextpred.html">torch_molecule.encoder.contextpred.modeling_contextpred</a></li>
-<li><a href="torch_molecule/encoder/edgepred/modeling_edgepred.html">torch_molecule.encoder.edgepred.modeling_edgepred</a></li>
-<li><a href="torch_molecule/encoder/moama/modeling_moama.html">torch_molecule.encoder.moama.modeling_moama</a></li>
-<li><a href="torch_molecule/encoder/supervised/modeling_supervised.html">torch_molecule.encoder.supervised.modeling_supervised</a></li>
-<li><a href="torch_molecule/generator/digress/modeling_digress.html">torch_molecule.generator.digress.modeling_digress</a></li>
-<li><a href="torch_molecule/generator/graph_dit/modeling_graph_dit.html">torch_molecule.generator.graph_dit.modeling_graph_dit</a></li>
-<li><a href="torch_molecule/generator/graph_ga/modeling_graph_ga.html">torch_molecule.generator.graph_ga.modeling_graph_ga</a></li>
-<li><a href="torch_molecule/generator/molgpt/modeling_molgpt.html">torch_molecule.generator.molgpt.modeling_molgpt</a></li>
 <li><a href="torch_molecule/nn/attention.html">torch_molecule.nn.attention</a></li>
 <li><a href="torch_molecule/nn/embedder.html">torch_molecule.nn.embedder</a></li>
 <li><a href="torch_molecule/nn/gnn.html">torch_molecule.nn.gnn</a></li>
 <li><a href="torch_molecule/nn/mlp.html">torch_molecule.nn.mlp</a></li>
-<li><a href="torch_molecule/predictor/dir/modeling_dir.html">torch_molecule.predictor.dir.modeling_dir</a></li>
 <li><a href="torch_molecule/predictor/gnn/modeling_gnn.html">torch_molecule.predictor.gnn.modeling_gnn</a></li>
 <li><a href="torch_molecule/predictor/grea/modeling_grea.html">torch_molecule.predictor.grea.modeling_grea</a></li>
 <li><a href="torch_molecule/predictor/irm/modeling_irm.html">torch_molecule.predictor.irm.modeling_irm</a></li>
-<li><a href="torch_molecule/predictor/lstm/modeling_lstm.html">torch_molecule.predictor.lstm.modeling_lstm</a></li>
-<li><a href="torch_molecule/predictor/rpgnn/modeling_rpgnn.html">torch_molecule.predictor.rpgnn.modeling_rpgnn</a></li>
 <li><a href="torch_molecule/predictor/sgir/modeling_sgir.html">torch_molecule.predictor.sgir.modeling_sgir</a></li>
-<li><a href="torch_molecule/predictor/smiles_transformer/modeling_transformer.html">torch_molecule.predictor.smiles_transformer.modeling_transformer</a></li>
-<li><a href="torch_molecule/predictor/ssr/modeling_ssr.html">torch_molecule.predictor.ssr.modeling_ssr</a></li>
 <li><a href="torch_molecule/utils/checker.html">torch_molecule.utils.checker</a></li>
 <li><a href="torch_molecule/utils/checkpoint.html">torch_molecule.utils.checkpoint</a></li>
 <li><a href="torch_molecule/utils/format.html">torch_molecule.utils.format</a></li>

_modules/torch_molecule/base/base.html

@@ -561,7 +561,7 @@ <h1>Source code for torch_molecule.base.base</h1><div class="highlight"><pre>
 <span class="sd">        &quot;&quot;&quot;</span>
         <span class="k">if</span> <span class="n">local_cache</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
             <span class="n">local_cache</span> <span class="o">=</span> <span class="s1">&#39;model.pt&#39;</span>
-        <span class="n">HuggingFaceCheckpointManager</span><span class="o">.</span><span class="n">load_model_from_hf</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">repo_id</span><span class="p">,</span> <span class="n">local_cache</span><span class="p">,</span> <span class="n">config_filename</span><span class="p">)</span></div>
+        <span class="n">HuggingFaceCheckpointManager</span><span class="o">.</span><span class="n">load_model_from_hf</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">repo_id</span><span class="p">,</span> <span class="n">local_cache</span><span class="p">,</span> <span class="n">config_filename</span><span class="o">=</span><span class="n">config_filename</span><span class="p">)</span></div>
 
 
 <div class="viewcode-block" id="BaseModel.save">