Update app.py

app.py

@@ -46,7 +46,7 @@ FTP_PASS = "GoogleBez12!"
 FTP_DIR = "1ink.us/stable_diff/"  # Remote directory on FTP server
 
 DESCRIPTIONXX = """
-    ## REALVISXL V5.0 BF16 ⚡⚡⚡⚡
+    ## ⚡⚡⚡⚡ REALVISXL V5.0 BF16 (Tester B) ⚡⚡⚡⚡
 """
 
 examples = [
@@ -163,10 +163,10 @@ def load_and_prepare_model(model_id):
     #pipe.vae.set_default_attn_processor()
     
     print(f'Pipeline: ')
-    print(f'_optional_components: {pipe._optional_components}')
-    print(f'watermark: {pipe.watermark}')
+    #print(f'_optional_components: {pipe._optional_components}')
+    #print(f'watermark: {pipe.watermark}')
     print(f'image_processor: {pipe.image_processor}')
-    print(f'feature_extractor: {pipe.feature_extractor}')
+    #print(f'feature_extractor: {pipe.feature_extractor}')
     print(f'init noise scale: {pipe.scheduler.init_noise_sigma}')
     #print(f'UNET: {pipe.unet}')
     pipe.watermark=None
@@ -210,6 +210,20 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
         seed = random.randint(0, MAX_SEED)
     return seed
 
+def uploadNote():
+    with open(filename, "w") as f:
+        f.write(f"Realvis 5.0 (Tester B): {seed} png\n")
+        f.write(f"Date/time: {timestamp} \n")
+        f.write(f"Prompt: {prompt} \n")
+        f.write(f"Steps: {num_inference_steps} \n")
+        f.write(f"Guidance Scale: {guidance_scale} \n")
+        f.write(f"SPACE SETUP: \n")
+        f.write(f"Use Model Dtype: no \n")
+        f.write(f"Model Scheduler: Euler_a custom before cuda \n")
+        f.write(f"Model VAE: default SG161222/RealVisXL_V5.0 \n")
+        f.write(f"Model UNET: default SG161222/RealVisXL_V5.0 \n")
+    upload_to_ftp(filename) 
+    
 @spaces.GPU(duration=30)
 def generate_30(
     model_choice: str,
@@ -253,25 +267,14 @@ def generate_30(
       # write note txt
     filename= f'tst_{seed}.txt'
     timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
-    with open(filename, "w") as f:
-        f.write(f"Realvis 5.0: {seed} png\n")
-        f.write(f"Date/time: {timestamp} \n")
-        f.write(f"Prompt: {prompt} \n")
-        f.write(f"Steps: {num_inference_steps} \n")
-        f.write(f"Guidance Scale: {guidance_scale} \n")
-        f.write(f"SPACE SETUP: \n")
-        f.write(f"Use Model Dtype: no \n")
-        f.write(f"Model Scheduler: Euler_a custom before cuda \n")
-        f.write(f"Model VAE: FP64 \n")
-        f.write(f"Model UNET: FP64 \n")
-    upload_to_ftp(filename) 
+
     for i in range(0, num_images, BATCH_SIZE):
         batch_options = options.copy()
         batch_options["prompt"] = options["prompt"][i:i+BATCH_SIZE]
         if "negative_prompt" in batch_options:
             batch_options["negative_prompt"] = options["negative_prompt"][i:i+BATCH_SIZE]
         images.extend(pipe(**batch_options).images)
-    sd_image_path = f"rv50_{seed}.png"
+    sd_image_path = f"rv50_B_{seed}.png"
     images[0].save(sd_image_path,optimize=False,compress_level=0)
     upload_to_ftp(sd_image_path)    
     image_paths = [save_image(img) for img in images]
@@ -322,25 +325,14 @@ def generate_60(
       # write note txt
     filename= f'tst_{seed}.txt'
     timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
-    with open(filename, "w") as f:
-        f.write(f"Realvis 5.0: {seed} png\n")
-        f.write(f"Date/time: {timestamp} \n")
-        f.write(f"Prompt: {prompt} \n")
-        f.write(f"Steps: {num_inference_steps} \n")
-        f.write(f"Guidance Scale: {guidance_scale} \n")
-        f.write(f"SPACE SETUP: \n")
-        f.write(f"Use Model Dtype: no \n")
-        f.write(f"Model Scheduler: Euler_a custom before cuda \n")
-        f.write(f"Model VAE: FP64 \n")
-        f.write(f"Model UNET: FP64 \n")
-    upload_to_ftp(filename) 
+    uploadNote()
     for i in range(0, num_images, BATCH_SIZE):
         batch_options = options.copy()
         batch_options["prompt"] = options["prompt"][i:i+BATCH_SIZE]
         if "negative_prompt" in batch_options:
             batch_options["negative_prompt"] = options["negative_prompt"][i:i+BATCH_SIZE]
         images.extend(pipe(**batch_options).images)
-    sd_image_path = f"rv50_{seed}.png"
+    sd_image_path = f"rv50_B_{seed}.png"
     images[0].save(sd_image_path,optimize=False,compress_level=0)
     upload_to_ftp(sd_image_path)    
     image_paths = [save_image(img) for img in images]
@@ -391,25 +383,14 @@ def generate_90(
       # write note txt
     filename= f'tst_{seed}.txt'
     timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
-    with open(filename, "w") as f:
-        f.write(f"Realvis 5.0: {seed} png\n")
-        f.write(f"Date/time: {timestamp} \n")
-        f.write(f"Prompt: {prompt} \n")
-        f.write(f"Steps: {num_inference_steps} \n")
-        f.write(f"Guidance Scale: {guidance_scale} \n")
-        f.write(f"SPACE SETUP: \n")
-        f.write(f"Use Model Dtype: no \n")
-        f.write(f"Model Scheduler: Euler_a custom before cuda \n")
-        f.write(f"Model VAE: FP64 \n")
-        f.write(f"Model UNET: FP64 \n")
-    upload_to_ftp(filename) 
+    uploadNote()
     for i in range(0, num_images, BATCH_SIZE):
         batch_options = options.copy()
         batch_options["prompt"] = options["prompt"][i:i+BATCH_SIZE]
         if "negative_prompt" in batch_options:
             batch_options["negative_prompt"] = options["negative_prompt"][i:i+BATCH_SIZE]
         images.extend(pipe(**batch_options).images)
-    sd_image_path = f"rv50_{seed}.png"
+    sd_image_path = f"rv50_B_{seed}.png"
     images[0].save(sd_image_path,optimize=False,compress_level=0)
     upload_to_ftp(sd_image_path)    
     image_paths = [save_image(img) for img in images]