alessandro trinca tornidor commited on
Commit
dfbc77d
·
1 Parent(s): 8959fb9

[fix] use sys.exit() instead than exit()

Browse files
Files changed (1) hide show
  1. model/llava/model/utils.py +3 -1
model/llava/model/utils.py CHANGED
@@ -1,3 +1,5 @@
 
 
1
  from transformers import AutoConfig
2
 
3
 
@@ -21,4 +23,4 @@ def auto_upgrade(config):
21
  print("Checkpoint upgraded.")
22
  else:
23
  print("Checkpoint upgrade aborted.")
24
- exit(1)
 
1
+ import sys
2
+
3
  from transformers import AutoConfig
4
 
5
 
 
23
  print("Checkpoint upgraded.")
24
  else:
25
  print("Checkpoint upgrade aborted.")
26
+ sys.exit(1)