michaelj commited on
Commit
685c49a
1 Parent(s): b7f7cdc

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. main.py +127 -127
main.py CHANGED
@@ -36,136 +36,136 @@ previous_height = 0
36
  previous_model_id = ""
37
  previous_num_of_images = 0
38
 
39
- parser = ArgumentParser(description=f"FAST SD CPU {constants.APP_VERSION}")
40
- parser.add_argument(
41
- "-s",
42
- "--share",
43
- action="store_true",
44
- help="Create sharable link(Web UI)",
45
- required=False,
46
- )
47
- group = parser.add_mutually_exclusive_group(required=False)
48
- group.add_argument(
49
- "-g",
50
- "--gui",
51
- action="store_true",
52
- help="Start desktop GUI",
53
- )
54
- group.add_argument(
55
- "-w",
56
- "--webui",
57
- action="store_true",
58
- help="Start Web UI",
59
- )
60
- group.add_argument(
61
- "-r",
62
- "--realtime",
63
- action="store_true",
64
- help="Start realtime inference UI(experimental)",
65
- )
66
- group.add_argument(
67
- "-v",
68
- "--version",
69
- action="store_true",
70
- help="Version",
71
- )
72
- parser.add_argument(
73
- "--lcm_model_id",
74
- type=str,
75
- help="Model ID or path,Default SimianLuo/LCM_Dreamshaper_v7",
76
- default="SimianLuo/LCM_Dreamshaper_v7",
77
- )
78
- parser.add_argument(
79
- "--prompt",
80
- type=str,
81
- help="Describe the image you want to generate",
82
- )
83
- parser.add_argument(
84
- "--image_height",
85
- type=int,
86
- help="Height of the image",
87
- default=512,
88
- )
89
- parser.add_argument(
90
- "--image_width",
91
- type=int,
92
- help="Width of the image",
93
- default=512,
94
- )
95
- parser.add_argument(
96
- "--inference_steps",
97
- type=int,
98
- help="Number of steps,default : 4",
99
- default=4,
100
- )
101
- parser.add_argument(
102
- "--guidance_scale",
103
- type=int,
104
- help="Guidance scale,default : 1.0",
105
- default=1.0,
106
- )
107
 
108
- parser.add_argument(
109
- "--number_of_images",
110
- type=int,
111
- help="Number of images to generate ,default : 1",
112
- default=1,
113
- )
114
- parser.add_argument(
115
- "--seed",
116
- type=int,
117
- help="Seed,default : -1 (disabled) ",
118
- default=-1,
119
- )
120
- parser.add_argument(
121
- "--use_openvino",
122
- action="store_true",
123
- help="Use OpenVINO model",
124
- )
125
 
126
- parser.add_argument(
127
- "--use_offline_model",
128
- action="store_true",
129
- help="Use offline model",
130
- )
131
- parser.add_argument(
132
- "--use_safety_checker",
133
- action="store_false",
134
- help="Use safety checker",
135
- )
136
- parser.add_argument(
137
- "--use_lcm_lora",
138
- action="store_true",
139
- help="Use LCM-LoRA",
140
- )
141
- parser.add_argument(
142
- "--base_model_id",
143
- type=str,
144
- help="LCM LoRA base model ID,Default Lykon/dreamshaper-8",
145
- default="Lykon/dreamshaper-8",
146
- )
147
- parser.add_argument(
148
- "--lcm_lora_id",
149
- type=str,
150
- help="LCM LoRA model ID,Default latent-consistency/lcm-lora-sdv1-5",
151
- default="latent-consistency/lcm-lora-sdv1-5",
152
- )
153
- parser.add_argument(
154
- "-i",
155
- "--interactive",
156
- action="store_true",
157
- help="Interactive CLI mode",
158
- )
159
- parser.add_argument(
160
- "--use_tiny_auto_encoder",
161
- action="store_true",
162
- help="Use tiny auto encoder for SD (TAESD)",
163
- )
164
- args = parser.parse_args()
165
 
166
- if args.version:
167
- print(APP_VERSION)
168
- exit()
169
 
170
  # parser.print_help()
171
  show_system_info()
 
36
  previous_model_id = ""
37
  previous_num_of_images = 0
38
 
39
+ # parser = ArgumentParser(description=f"FAST SD CPU {constants.APP_VERSION}")
40
+ # parser.add_argument(
41
+ # "-s",
42
+ # "--share",
43
+ # action="store_true",
44
+ # help="Create sharable link(Web UI)",
45
+ # required=False,
46
+ # )
47
+ # group = parser.add_mutually_exclusive_group(required=False)
48
+ # group.add_argument(
49
+ # "-g",
50
+ # "--gui",
51
+ # action="store_true",
52
+ # help="Start desktop GUI",
53
+ # )
54
+ # group.add_argument(
55
+ # "-w",
56
+ # "--webui",
57
+ # action="store_true",
58
+ # help="Start Web UI",
59
+ # )
60
+ # group.add_argument(
61
+ # "-r",
62
+ # "--realtime",
63
+ # action="store_true",
64
+ # help="Start realtime inference UI(experimental)",
65
+ # )
66
+ # group.add_argument(
67
+ # "-v",
68
+ # "--version",
69
+ # action="store_true",
70
+ # help="Version",
71
+ # )
72
+ # parser.add_argument(
73
+ # "--lcm_model_id",
74
+ # type=str,
75
+ # help="Model ID or path,Default SimianLuo/LCM_Dreamshaper_v7",
76
+ # default="SimianLuo/LCM_Dreamshaper_v7",
77
+ # )
78
+ # parser.add_argument(
79
+ # "--prompt",
80
+ # type=str,
81
+ # help="Describe the image you want to generate",
82
+ # )
83
+ # parser.add_argument(
84
+ # "--image_height",
85
+ # type=int,
86
+ # help="Height of the image",
87
+ # default=512,
88
+ # )
89
+ # parser.add_argument(
90
+ # "--image_width",
91
+ # type=int,
92
+ # help="Width of the image",
93
+ # default=512,
94
+ # )
95
+ # parser.add_argument(
96
+ # "--inference_steps",
97
+ # type=int,
98
+ # help="Number of steps,default : 4",
99
+ # default=4,
100
+ # )
101
+ # parser.add_argument(
102
+ # "--guidance_scale",
103
+ # type=int,
104
+ # help="Guidance scale,default : 1.0",
105
+ # default=1.0,
106
+ # )
107
 
108
+ # parser.add_argument(
109
+ # "--number_of_images",
110
+ # type=int,
111
+ # help="Number of images to generate ,default : 1",
112
+ # default=1,
113
+ # )
114
+ # parser.add_argument(
115
+ # "--seed",
116
+ # type=int,
117
+ # help="Seed,default : -1 (disabled) ",
118
+ # default=-1,
119
+ # )
120
+ # parser.add_argument(
121
+ # "--use_openvino",
122
+ # action="store_true",
123
+ # help="Use OpenVINO model",
124
+ # )
125
 
126
+ # parser.add_argument(
127
+ # "--use_offline_model",
128
+ # action="store_true",
129
+ # help="Use offline model",
130
+ # )
131
+ # parser.add_argument(
132
+ # "--use_safety_checker",
133
+ # action="store_false",
134
+ # help="Use safety checker",
135
+ # )
136
+ # parser.add_argument(
137
+ # "--use_lcm_lora",
138
+ # action="store_true",
139
+ # help="Use LCM-LoRA",
140
+ # )
141
+ # parser.add_argument(
142
+ # "--base_model_id",
143
+ # type=str,
144
+ # help="LCM LoRA base model ID,Default Lykon/dreamshaper-8",
145
+ # default="Lykon/dreamshaper-8",
146
+ # )
147
+ # parser.add_argument(
148
+ # "--lcm_lora_id",
149
+ # type=str,
150
+ # help="LCM LoRA model ID,Default latent-consistency/lcm-lora-sdv1-5",
151
+ # default="latent-consistency/lcm-lora-sdv1-5",
152
+ # )
153
+ # parser.add_argument(
154
+ # "-i",
155
+ # "--interactive",
156
+ # action="store_true",
157
+ # help="Interactive CLI mode",
158
+ # )
159
+ # parser.add_argument(
160
+ # "--use_tiny_auto_encoder",
161
+ # action="store_true",
162
+ # help="Use tiny auto encoder for SD (TAESD)",
163
+ # )
164
+ # args = parser.parse_args()
165
 
166
+ # if args.version:
167
+ # print(APP_VERSION)
168
+ # exit()
169
 
170
  # parser.print_help()
171
  show_system_info()