Enable embedding projector to show for input embeddings.

PiperOrigin-RevId: 480040598
PAIR-code · Oct 10, 2022 · e7ac98b · e7ac98b
1 parent 3c0929f
commit e7ac98b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lit_nlp/client/modules/embeddings_module.ts b/lit_nlp/client/modules/embeddings_module.ts
@@ -697,6 +697,9 @@ export class EmbeddingsModule extends LitModule {
   }
 
   static override shouldDisplayModule(modelSpecs: ModelInfoMap, datasetSpec: Spec) {
+    // Check if there are Embeddings in the input data.
+    if (findSpecKeys(datasetSpec, Embeddings).length > 0) return true;
+
     // Ensure there are embeddings to use and that projection interpreters
     // are loaded.
     if (!doesOutputSpecContain(modelSpecs, Embeddings)) {