Fix missing position ids

This commit is contained in:
silveroxides 2026-06-20 18:08:55 +02:00
parent b45f5de683
commit 6e434c785b

View File

@ -253,8 +253,8 @@ class Ideogram4Transformer2DModel(Ideogram4Transformer):
for ref in ref_latents:
ref_b, ref_c, ref_h, ref_w = ref.shape
ref_gh = ref_h // self.patch_size
ref_gw = ref_w // self.patch_size
ref_gh = ref_h
ref_gw = ref_w
if index_ref_method:
index += 1
@ -343,8 +343,8 @@ class Ideogram4Transformer2DModel(Ideogram4Transformer):
for ref in ref_latents:
ref_b, ref_c, ref_h, ref_w = ref.shape
ref_gh = ref_h // self.patch_size
ref_gw = ref_w // self.patch_size
ref_gh = ref_h
ref_gw = ref_w
if index_ref_method:
index += 1