nodes/Junst-ComfyUI-Concept-Diffusion
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix context parameter duplication and remove debug code

Browse Source 
- Remove debug logging that shows model structure
- Fix 'got multiple values for keyword argument context' error
- Use positional arguments instead of keyword arguments for apply_model
- Clean up excessive debug output
- Improve error handling without verbose logging
This commit is contained in:
Junst
2025-09-24 16:24:43 +09:00
parent 3770b08735
commit 9b0aa20e90
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
concept_attention.py
+3 -3
View File
@@ -84,16 +84,16 @@ class ConceptAttention:
with torch.no_grad():
# Use the model's apply_model method (ComfyUI ModelPatcher)
if hasattr(self.model, 'apply_model'):
output = self.model.apply_model(x, timestep_tensor, context=context, y=y)
output = self.model.apply_model(x, timestep_tensor, context, y)
elif hasattr(self.model, 'model') and hasattr(self.model.model, 'apply_model'):
# Nested model access
output = self.model.model.apply_model(x, timestep_tensor, context=context, y=y)
output = self.model.model.apply_model(x, timestep_tensor, context, y)
else:
# Try to access the underlying diffusion model
if hasattr(self.model, 'model'):
diffusion_model = self.model.model
if hasattr(diffusion_model, 'apply_model'):
output = diffusion_model.apply_model(x, timestep_tensor, context=context, y=y)
output = diffusion_model.apply_model(x, timestep_tensor, context, y)
else:
raise RuntimeError("Cannot find apply_model method in model")
else: