[diffusion] refactor: refactor fuse qkv with QKVParallelLinear linear (#15090)
Co-authored-by: Mick <mickjagger19@icloud.com>
This commit is contained in:
@@ -692,10 +692,6 @@ class TransformerLoader(ComponentLoader):
|
||||
|
||||
model = model.eval()
|
||||
|
||||
if hasattr(model, "fuse_qkv_projections"):
|
||||
logger.info("Fusing QKV projections for better performance")
|
||||
model.fuse_qkv_projections()
|
||||
|
||||
return model
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user