This commit is contained in:
Ryan Dick 2023-09-08 18:05:31 -04:00
parent a255624984
commit 50a0691514
2 changed files with 3 additions and 5 deletions

View File

@ -1,9 +1,7 @@
from __future__ import annotations from __future__ import annotations
import dataclasses
import inspect
from contextlib import nullcontext from contextlib import nullcontext
from dataclasses import dataclass, field from dataclasses import dataclass
from typing import Any, Callable, List, Optional, Union from typing import Any, Callable, List, Optional, Union
import einops import einops

View File

@ -55,11 +55,11 @@ class PostprocessingSettings:
class IPAdapterConditioningInfo: class IPAdapterConditioningInfo:
cond_image_prompt_embeds: torch.Tensor cond_image_prompt_embeds: torch.Tensor
"""IP-Adapter image encoder conditioning embeddings. """IP-Adapter image encoder conditioning embeddings.
Shape: (batch_size, num_tokens, encoding_dim). Typically: (1, 4, 1024) TODO(ryand): confirm Shape: (batch_size, num_tokens, encoding_dim).
""" """
uncond_image_prompt_embeds: torch.Tensor uncond_image_prompt_embeds: torch.Tensor
"""IP-Adapter image encoding embeddings to use for unconditional generation. """IP-Adapter image encoding embeddings to use for unconditional generation.
Shape: (batch_size, num_tokens, encoding_dim). Typically: (1, 4, 1024) TODO(ryand): confirm Shape: (batch_size, num_tokens, encoding_dim).
""" """