AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Code Activity

remove debug logs

Browse Source
This commit is contained in:
Aryan
2025-04-01 01:39:00 +02:00
parent dd69b41834
commit 7ab424a15a
1 changed files with 0 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/diffusers/hooks/first_block_cache.py
View File

@@ -87,8 +87,6 @@ class FBCHeadBlockHook(ModelHook):
if not should_compute:
# Apply caching
logger.info("Skipping forward pass through remaining blocks")
if is_output_tuple:
hs = self.shared_state.tail_block_residuals[0] + output[self._metadata.return_hidden_states_index]
else:
@@ -109,7 +107,6 @@ class FBCHeadBlockHook(ModelHook):
return_output = hs
return return_output
else:
logger.info("Computing forward pass through remaining blocks")
if is_output_tuple:
head_block_output = [None] * len(output)
head_block_output[0] = output[self._metadata.return_hidden_states_index]
@@ -131,7 +128,6 @@ class FBCHeadBlockHook(ModelHook):
hs_absmean = (hs_residual - prev_hs_residual).abs().mean()
prev_hs_mean = prev_hs_residual.abs().mean()
diff = (hs_absmean / prev_hs_mean).item()
logger.info(f"Diff: {diff}, Threshold: {self.threshold}")
return diff > self.threshold