Update prompt_encoder/encoder.py

prompt_encoder/encoder.py

@@ -82,7 +82,7 @@ class MaterialPromptEncoder(BasePromptEncoder):
         for prompt in prompt:
             if isinstance(prompt, str):
                 embs.append(self.encode_text(prompt))
-            elif type(prompt, get_args(ImageInput)):
+            elif type(prompt) in get_args(ImageInput)):
                 embs.append(self.encode_image(prompt))
             else:
                 raise NotImplementedError