Update geneformer/emb_extractor.py (#303)

- Update geneformer/emb_extractor.py (d1f49b5321b8f1b9709ddd6b27a577f6acc0ddeb)


Co-authored-by: Han Chen <[email protected]>

geneformer/emb_extractor.py

@@ -565,7 +565,7 @@ class EmbExtractor:
                 filtered_input_data, cell_state, self.nproc
             )
         downsampled_data = pu.downsample_and_sort(filtered_input_data, self.max_ncells)
-        model = pu.load_model(self.model_type, self.num_classes, model_directory)
+        model = pu.load_model(self.model_type, self.num_classes, model_directory, mode = "eval")
         layer_to_quant = pu.quant_layers(model) + self.emb_layer
         embs = get_embs(
             model,