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
invokeai/backend/stable_diffusion

@ -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

@ -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).
"""