thefreeham commited on
Commit
232330c
1 Parent(s): 3194719

Create utils.py

Browse files
Files changed (1) hide show
  1. utils.py +16 -0
utils.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from consts import ModelSize
2
+
3
+
4
+ def parse_arg_boolean(value):
5
+ value = value.lower()
6
+
7
+ if value in ["true", "yes", "y", "1", "t"]:
8
+ return True
9
+ elif value in ["false", "no", "n", "0", "f"]:
10
+ return False
11
+
12
+ return False
13
+
14
+ def parse_arg_dalle_version(value):
15
+ value = value.lower()
16
+ return ModelSize[value.upper()]