Skip to content

Commit f18ccbd

Browse files
committed
Make exception RuntimeError specific
Signed-off-by: Timur Rvachov <[email protected]>
1 parent 60c2ce5 commit f18ccbd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

nemo/lightning/io/api.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -270,7 +270,7 @@ def export_ckpt(
270270
modelopt_export_kwargs = modelopt_export_kwargs or {}
271271
# First try to export via ModelOpt route. If rejected, return to the default route
272272
output = export_hf_checkpoint(path, _output_path, **modelopt_export_kwargs)
273-
except:
273+
except RuntimeError:
274274
output = None
275275
if output is not None:
276276
return output

0 commit comments

Comments
 (0)