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
import dataclasses
import inspect
from contextlib import nullcontext
from dataclasses import dataclass, field
from dataclasses import dataclass
from typing import Any, Callable, List, Optional, Union
import einops

View File

@ -55,11 +55,11 @@ class PostprocessingSettings:
class IPAdapterConditioningInfo:
cond_image_prompt_embeds: torch.Tensor
"""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
"""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).
"""