File tree Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Original file line number Diff line number Diff line change @@ -1425,21 +1425,21 @@ class name matches will be compiled.
1425
1425
`_no_split_modules`.
1426
1426
1427
1427
Once discovered, each matching sub-module is compiled by calling
1428
- `` submodule.compile(*args, **kwargs)` `. Any positional or keyword
1429
- arguments you supply to :meth: `compile_repeated_blocks` are forwarded
1428
+ `submodule.compile(*args, **kwargs)`. Any positional or keyword
1429
+ arguments you supply to `compile_repeated_blocks` are forwarded
1430
1430
verbatim to `torch.compile`.
1431
1431
"""
1432
1432
repeated_blocks = getattr (self , "_repeated_blocks" , None )
1433
1433
1434
1434
if not repeated_blocks :
1435
- logger .warning ("_repeated_blocks attribute is empty. Using _no_split_modules to find compile regions." )
1435
+ logger .warning ("` _repeated_blocks` attribute is empty. Using ` _no_split_modules` to find compile regions." )
1436
1436
1437
1437
repeated_blocks = getattr (self , "_no_split_modules" , None )
1438
1438
1439
1439
if not repeated_blocks :
1440
1440
raise ValueError (
1441
- "Both _repeated_blocks and _no_split_modules attribute are empty. "
1442
- "Set _repeated_blocks for the model to benefit from faster compilation. "
1441
+ "Both ` _repeated_blocks` and ` _no_split_modules` attribute are empty. "
1442
+ "Set ` _repeated_blocks` for the model to benefit from faster compilation. "
1443
1443
)
1444
1444
1445
1445
has_compiled_region = False
You can’t perform that action at this time.
0 commit comments