(self)
| 89 | return self.multigpu_clones.get(device, self) |
| 90 | |
| 91 | def init_proj(self): |
| 92 | image_proj_model = ImageProjModel( |
| 93 | cross_attention_dim=self.cross_attention_dim, |
| 94 | clip_embeddings_dim=self.clip_embeddings_dim, |
| 95 | clip_extra_context_tokens=self.clip_extra_context_tokens |
| 96 | ) |
| 97 | return image_proj_model |
| 98 | |
| 99 | def init_proj_plus(self): |
| 100 | image_proj_model = Resampler( |