b

app_v3.py

@@ -21,9 +21,11 @@ from threading import Thread
 from typing import Generator
 # from peft import PeftModel, PeftConfig
 import gradio as gr
-from huggingface_hub import CommitScheduler, HfApi
+from huggingface_hub import CommitScheduler, HfApi, logging
 from debug import log_params, scheduler
 
+logging.set_verbosity_debug()
+
 huggingface_token = os.getenv("HUGGINFACE_TOKEN")
 MAX_SEED = 1000000
 

debug.py

@@ -32,7 +32,7 @@ def save_image(image_id, image_path: Image.Image) -> None:
     os.makedirs(IMAGE_DIR, exist_ok=True)
     print("Image ID: " + image_id)
     print("Image ID Type: " + str(type(image_id)))
-    save_image_path = os.path.join(IMAGE_DIR, f"{image_id}.png")
+    save_image_path = os.path.join(IMAGE_DIR, f"{image_id}")
     print("Save image path: " + save_image_path)
     with scheduler.lock:
         try: