We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 60c2ce5 commit f18ccbdCopy full SHA for f18ccbd
nemo/lightning/io/api.py
@@ -270,7 +270,7 @@ def export_ckpt(
270
modelopt_export_kwargs = modelopt_export_kwargs or {}
271
# First try to export via ModelOpt route. If rejected, return to the default route
272
output = export_hf_checkpoint(path, _output_path, **modelopt_export_kwargs)
273
- except:
+ except RuntimeError:
274
output = None
275
if output is not None:
276
return output
0 commit comments