Update prompt_encoder/encoder.py
Browse files
prompt_encoder/encoder.py
CHANGED
@@ -82,7 +82,7 @@ class MaterialPromptEncoder(BasePromptEncoder):
|
|
82 |
for prompt in prompt:
|
83 |
if isinstance(prompt, str):
|
84 |
embs.append(self.encode_text(prompt))
|
85 |
-
elif type(prompt) in get_args(ImageInput)
|
86 |
embs.append(self.encode_image(prompt))
|
87 |
else:
|
88 |
raise NotImplementedError
|
|
|
82 |
for prompt in prompt:
|
83 |
if isinstance(prompt, str):
|
84 |
embs.append(self.encode_text(prompt))
|
85 |
+
elif type(prompt) in get_args(ImageInput):
|
86 |
embs.append(self.encode_image(prompt))
|
87 |
else:
|
88 |
raise NotImplementedError
|