gvecchio commited on
Commit
7e6c409
1 Parent(s): 95a98f1

Update prompt_encoder/encoder.py

Browse files
Files changed (1) hide show
  1. prompt_encoder/encoder.py +1 -1
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