Skip to content

Commit 069f86a

Browse files
committed
Manual fixes before merge
1 parent 4fac443 commit 069f86a

File tree

2 files changed

+7
-0
lines changed

2 files changed

+7
-0
lines changed

QEfficient/generation/text_generation_inference.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -809,12 +809,14 @@ def run_prefill(self, prompt, generation_len, prefill_logit_bs=1, decode_batch_i
809809
self.list_of_comp_ctx_lengths_prefill = [np.zeros(length) for length in self.comp_ctx_lengths_prefill]
810810
prefill_ccl_id = 0
811811
inputs["comp_ctx_lengths"] = self.list_of_comp_ctx_lengths_prefill[prefill_ccl_id]
812+
print(f"CCL Prefill: {self.comp_ctx_lengths_prefill[prefill_ccl_id]}")
812813

813814
for i in range(num_chunks):
814815
if self.comp_ctx_lengths_prefill is not None:
815816
if (i + 1) * self._prefill_seq_len > self.comp_ctx_lengths_prefill[prefill_ccl_id]:
816817
prefill_ccl_id = min(prefill_ccl_id + 1, len(self.comp_ctx_lengths_prefill) - 1)
817818
inputs["comp_ctx_lengths"] = self.list_of_comp_ctx_lengths_prefill[prefill_ccl_id]
819+
print(f"CCL Prefill: {self.comp_ctx_lengths_prefill[prefill_ccl_id]}")
818820

819821
chunk_inputs = inputs.copy()
820822
chunk_inputs["input_ids"] = inputs["input_ids"][

QEfficient/utils/constants.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -125,6 +125,10 @@ def get_models_dir():
125125
# Wav2Vec2 Constant
126126
WAV2VEC2_MAX_SEQ_LEN = 480000 # 30 seconds of audio at 16 kHz sampling rate (16,000 samples/sec × 30 sec)
127127

128+
# Qwen2_5_vl Constants
129+
QWEN2_5_VL_HEIGHT = 354
130+
QWEN2_5_VL_WIDTH = 536
131+
128132

129133
class Constants:
130134
# Export Constants.
@@ -235,3 +239,4 @@ class QnnConstants:
235239
},
236240
"SKIP_QNN_CONVERTER_STEP": False,
237241
}
242+

0 commit comments

Comments
 (0)