Skip to content

Commit 0be4cea

Browse files
authored
fix image token typo
1 parent 106cf26 commit 0be4cea

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

fastcomposer/pipeline.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -244,7 +244,7 @@ def __call__(
244244
assert do_classifier_free_guidance
245245

246246
# 3. Encode input prompt
247-
prompt_text_only = prompt.replace("<image>", "")
247+
prompt_text_only = prompt.replace("img", "")
248248

249249
prompt_embeds = self._encode_prompt(
250250
prompt_text_only,

0 commit comments

Comments
 (0)