ford442 commited on
Commit
e89aac5
·
verified ·
1 Parent(s): be666f7

Update ip_adapter/ip_adapter.py

Browse files
Files changed (1) hide show
  1. ip_adapter/ip_adapter.py +2 -2
ip_adapter/ip_adapter.py CHANGED
@@ -25,7 +25,7 @@ class ImageProjModel(torch.nn.Module):
25
  clip_extra_context_tokens = self.proj(embeds).reshape(-1, self.clip_extra_context_tokens, self.cross_attention_dim)
26
  clip_extra_context_tokens = self.norm(clip_extra_context_tokens)
27
  return clip_extra_context_tokens
28
- '''
29
  class IPAdapter:
30
 
31
  def __init__(self, sd_pipe, image_encoder_path, ip_ckpt, device, num_tokens=4):
@@ -180,7 +180,6 @@ class IPAdapter:
180
 
181
  return images
182
 
183
- '''
184
  class IPAdapterXL(IPAdapter):
185
  """SDXL"""
186
 
@@ -285,6 +284,7 @@ class IPAdapterXL(IPAdapter):
285
  negative_prompt_embeds = torch.cat([negative_prompt_embeds, uncond_image_prompt_embeds], dim=1)
286
 
287
  generator = torch.Generator(self.device).manual_seed(seed) if seed is not None else None
 
288
  images = self.pipe(
289
  prompt_embeds=prompt_embeds,
290
  negative_prompt_embeds=negative_prompt_embeds,
 
25
  clip_extra_context_tokens = self.proj(embeds).reshape(-1, self.clip_extra_context_tokens, self.cross_attention_dim)
26
  clip_extra_context_tokens = self.norm(clip_extra_context_tokens)
27
  return clip_extra_context_tokens
28
+
29
  class IPAdapter:
30
 
31
  def __init__(self, sd_pipe, image_encoder_path, ip_ckpt, device, num_tokens=4):
 
180
 
181
  return images
182
 
 
183
  class IPAdapterXL(IPAdapter):
184
  """SDXL"""
185
 
 
284
  negative_prompt_embeds = torch.cat([negative_prompt_embeds, uncond_image_prompt_embeds], dim=1)
285
 
286
  generator = torch.Generator(self.device).manual_seed(seed) if seed is not None else None
287
+
288
  images = self.pipe(
289
  prompt_embeds=prompt_embeds,
290
  negative_prompt_embeds=negative_prompt_embeds,