update

hf_demo.py

@@ -116,10 +116,10 @@ with block:
         with gr.Row():
             text = gr.Textbox(
                 label="Enter your prompt (long and detailed would be better):",
-                max_lines=2,
+                max_lines=10,
                 placeholder="Enter your prompt (long and detailed would be better)",
                 container=True,
-                value="Park with cherry blossom trees, picnicker’s and a clear blue pond.",
+                value="A blue bench situated in a park, surrounded by trees and leaves. The bench is positioned under a tree, providing shade and a peaceful atmosphere. There are several benches in the park, with one being closer to the foreground and the others further in the background. A person can be seen in the distance, possibly enjoying the park or taking a walk. The overall scene is serene and inviting, with the bench serving as a focal point in the park's landscape.",
             )
 
         with gr.Tab('Generation'):
@@ -165,6 +165,7 @@ with block:
                         elem_id="gallery",
                         height="auto",
                         scale=1,
+                        value="data/003904765.jpg"
                     )
                 with gr.Column(scale=2):
                     with gr.Row():
@@ -193,8 +194,8 @@ with block:
                             )
                     start_timestep = gr.Slider(label="Adapter Timestep", minimum=0, maximum=1000, value=800, step=1)
             with gr.Row():
-                btn_style_ori = gr.Button("Art-Free Stylization", scale=1)
-                btn_style_art = gr.Button("Artistic Stylization", scale=1)
+                btn_style_ori = gr.Button("Art-Free Stylize", scale=1)
+                btn_style_art = gr.Button("Artistic Stylize", scale=1)
 
 
         with gr.Row():

inference.py

@@ -38,6 +38,7 @@ from custom_datasets import get_dataset
 from diffusers.utils.torch_utils import randn_tensor
 import pickle
 import time
+from datetime import datetime
 def flush():
     torch.cuda.empty_cache()
     gc.collect()
@@ -233,6 +234,7 @@ def inference(network: LoRANetwork, tokenizer: CLIPTokenizer, text_encoder: CLIP
               dataloader, height:int, width:int, scales:list = np.linspace(0,2,5),save_dir:str=None, seed:int = None,
               weight_dtype: torch.dtype = torch.float32, device: torch.device="cuda", batch_size:int=1, steps:int=50, guidance_scale:float=7.5, start_noise:int=800,
               uncond_prompt:str=None, uncond_embed=None, style_prompt = None, show:bool = False, no_load:bool=False, from_scratch=False):
+    print(f"current time: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
     print(f"save dir: {save_dir}")
     if start_noise < 0:
         assert from_scratch