Upload folder using huggingface_hub

app.py

@@ -36,138 +36,138 @@ previous_height = 0
 previous_model_id = ""
 previous_num_of_images = 0
 
-# parser = ArgumentParser(description=f"FAST SD CPU {constants.APP_VERSION}")
-# parser.add_argument(
-#     "-s",
-#     "--share",
-#     action="store_true",
-#     help="Create sharable link(Web UI)",
-#     required=False,
-# )
-# group = parser.add_mutually_exclusive_group(required=False)
-# group.add_argument(
-#     "-g",
-#     "--gui",
-#     action="store_true",
-#     help="Start desktop GUI",
-# )
-# group.add_argument(
-#     "-w",
-#     "--webui",
-#     action="store_true",
-#     help="Start Web UI",
-# )
-# group.add_argument(
-#     "-r",
-#     "--realtime",
-#     action="store_true",
-#     help="Start realtime inference UI(experimental)",
-# )
-# group.add_argument(
-#     "-v",
-#     "--version",
-#     action="store_true",
-#     help="Version",
-# )
-# parser.add_argument(
-#     "--lcm_model_id",
-#     type=str,
-#     help="Model ID or path,Default SimianLuo/LCM_Dreamshaper_v7",
-#     default="SimianLuo/LCM_Dreamshaper_v7",
-# )
-# parser.add_argument(
-#     "--prompt",
-#     type=str,
-#     help="Describe the image you want to generate",
-# )
-# parser.add_argument(
-#     "--image_height",
-#     type=int,
-#     help="Height of the image",
-#     default=512,
-# )
-# parser.add_argument(
-#     "--image_width",
-#     type=int,
-#     help="Width of the image",
-#     default=512,
-# )
-# parser.add_argument(
-#     "--inference_steps",
-#     type=int,
-#     help="Number of steps,default : 4",
-#     default=4,
-# )
-# parser.add_argument(
-#     "--guidance_scale",
-#     type=int,
-#     help="Guidance scale,default : 1.0",
-#     default=1.0,
-# )
+parser = ArgumentParser(description=f"FAST SD CPU {constants.APP_VERSION}")
+parser.add_argument(
+    "-s",
+    "--share",
+    action="store_true",
+    help="Create sharable link(Web UI)",
+    required=False,
+)
+group = parser.add_mutually_exclusive_group(required=False)
+group.add_argument(
+    "-g",
+    "--gui",
+    action="store_true",
+    help="Start desktop GUI",
+)
+group.add_argument(
+    "-w",
+    "--webui",
+    action="store_true",
+    help="Start Web UI",
+)
+group.add_argument(
+    "-r",
+    "--realtime",
+    action="store_true",
+    help="Start realtime inference UI(experimental)",
+)
+group.add_argument(
+    "-v",
+    "--version",
+    action="store_true",
+    help="Version",
+)
+parser.add_argument(
+    "--lcm_model_id",
+    type=str,
+    help="Model ID or path,Default SimianLuo/LCM_Dreamshaper_v7",
+    default="SimianLuo/LCM_Dreamshaper_v7",
+)
+parser.add_argument(
+    "--prompt",
+    type=str,
+    help="Describe the image you want to generate",
+)
+parser.add_argument(
+    "--image_height",
+    type=int,
+    help="Height of the image",
+    default=512,
+)
+parser.add_argument(
+    "--image_width",
+    type=int,
+    help="Width of the image",
+    default=512,
+)
+parser.add_argument(
+    "--inference_steps",
+    type=int,
+    help="Number of steps,default : 4",
+    default=4,
+)
+parser.add_argument(
+    "--guidance_scale",
+    type=int,
+    help="Guidance scale,default : 1.0",
+    default=1.0,
+)
 
-# parser.add_argument(
-#     "--number_of_images",
-#     type=int,
-#     help="Number of images to generate ,default : 1",
-#     default=1,
-# )
-# parser.add_argument(
-#     "--seed",
-#     type=int,
-#     help="Seed,default : -1 (disabled) ",
-#     default=-1,
-# )
-# parser.add_argument(
-#     "--use_openvino",
-#     action="store_true",
-#     help="Use OpenVINO model",
-# )
+parser.add_argument(
+    "--number_of_images",
+    type=int,
+    help="Number of images to generate ,default : 1",
+    default=1,
+)
+parser.add_argument(
+    "--seed",
+    type=int,
+    help="Seed,default : -1 (disabled) ",
+    default=-1,
+)
+parser.add_argument(
+    "--use_openvino",
+    action="store_true",
+    help="Use OpenVINO model",
+)
 
-# parser.add_argument(
-#     "--use_offline_model",
-#     action="store_true",
-#     help="Use offline model",
-# )
-# parser.add_argument(
-#     "--use_safety_checker",
-#     action="store_false",
-#     help="Use safety checker",
-# )
-# parser.add_argument(
-#     "--use_lcm_lora",
-#     action="store_true",
-#     help="Use LCM-LoRA",
-# )
-# parser.add_argument(
-#     "--base_model_id",
-#     type=str,
-#     help="LCM LoRA base model ID,Default Lykon/dreamshaper-8",
-#     default="Lykon/dreamshaper-8",
-# )
-# parser.add_argument(
-#     "--lcm_lora_id",
-#     type=str,
-#     help="LCM LoRA model ID,Default latent-consistency/lcm-lora-sdv1-5",
-#     default="latent-consistency/lcm-lora-sdv1-5",
-# )
-# parser.add_argument(
-#     "-i",
-#     "--interactive",
-#     action="store_true",
-#     help="Interactive CLI mode",
-# )
-# parser.add_argument(
-#     "--use_tiny_auto_encoder",
-#     action="store_true",
-#     help="Use tiny auto encoder for SD (TAESD)",
-# )
-# args = parser.parse_args()
+parser.add_argument(
+    "--use_offline_model",
+    action="store_true",
+    help="Use offline model",
+)
+parser.add_argument(
+    "--use_safety_checker",
+    action="store_false",
+    help="Use safety checker",
+)
+parser.add_argument(
+    "--use_lcm_lora",
+    action="store_true",
+    help="Use LCM-LoRA",
+)
+parser.add_argument(
+    "--base_model_id",
+    type=str,
+    help="LCM LoRA base model ID,Default Lykon/dreamshaper-8",
+    default="Lykon/dreamshaper-8",
+)
+parser.add_argument(
+    "--lcm_lora_id",
+    type=str,
+    help="LCM LoRA model ID,Default latent-consistency/lcm-lora-sdv1-5",
+    default="latent-consistency/lcm-lora-sdv1-5",
+)
+parser.add_argument(
+    "-i",
+    "--interactive",
+    action="store_true",
+    help="Interactive CLI mode",
+)
+parser.add_argument(
+    "--use_tiny_auto_encoder",
+    action="store_true",
+    help="Use tiny auto encoder for SD (TAESD)",
+)
+args = parser.parse_args()
 
-# if args.version:
-#     print(APP_VERSION)
-#     exit()
+if args.version:
+    print(APP_VERSION)
+    exit()
 
-# parser.print_help()
+parser.print_help()
 show_system_info()
 print(f"Using device : {constants.DEVICE}")
 app_settings = get_settings()