update candidate checkpoints.

app.py

@@ -74,7 +74,7 @@ def process_multi_wrapper_only_show_rendered(rendered_txt_0, rendered_txt_1, ren
                                      shared_eta, shared_a_prompt, shared_n_prompt, 
                                      only_show_rendered_image=True)  
 
-def load_ckpt(model_ckpt = "LAION-Glyph-10M"):
+def load_ckpt(model_ckpt = "LAION-Glyph-10M-Epoch-5"):
     global render_tool, model
     if torch.cuda.is_available():
         for i in range(5):
@@ -84,8 +84,11 @@ def load_ckpt(model_ckpt = "LAION-Glyph-10M"):
 
     if model_ckpt == "LAION-Glyph-1M":
         model = load_model_ckpt(model, "laion1M_model_wo_ema.ckpt")
-    elif model_ckpt == "LAION-Glyph-10M":
-        model = load_model_ckpt(model, "model_wo_ema.ckpt")
+    elif model_ckpt == "LAION-Glyph-10M-Epoch-5":
+        model = load_model_ckpt(model, "laion10M_epoch_5_model_wo_ema.ckpt")
+    elif model_ckpt == "LAION-Glyph-10M-Epoch-6":
+        model = load_model_ckpt(model, "laion10M_epoch_6_model_wo_ema.ckpt")
+
     render_tool = Render_Text(model)
     output_str = f"already change the model checkpoint to {model_ckpt}"
     print(output_str)
@@ -96,7 +99,8 @@ def load_ckpt(model_ckpt = "LAION-Glyph-10M"):
         print("empty the cuda cache")
 
 cfg = OmegaConf.load("config.yaml")
-model = load_model_from_config(cfg, "model_wo_ema.ckpt", verbose=True)
+model = load_model_from_config(cfg, "laion10M_epoch_6_model_wo_ema.ckpt", verbose=True)
+# model = load_model_from_config(cfg, "model_wo_ema.ckpt", verbose=True)
 # model = load_model_from_config(cfg, "model_states.pt", verbose=True)
 # model = load_model_from_config(cfg, "model.ckpt", verbose=True)
 # ddim_sampler = DDIMSampler(model)
@@ -148,7 +152,7 @@ with block:
                 with gr.Accordion("Model Options", open=False):
                     with gr.Row():
                         # model_ckpt = gr.inputs.Dropdown(["LAION-Glyph-10M", "Textcaps5K-10"], label="Checkpoint", default = "LAION-Glyph-10M")
-                        model_ckpt = gr.inputs.Dropdown(["LAION-Glyph-10M", "LAION-Glyph-1M"], label="Checkpoint", default = "LAION-Glyph-10M")
+                        model_ckpt = gr.inputs.Dropdown(["LAION-Glyph-10M-Epoch-6", "LAION-Glyph-10M-Epoch-5", "LAION-Glyph-1M"], label="Checkpoint", default = "LAION-Glyph-10M-Epoch-6")
                         load_button = gr.Button(value = "Load Checkpoint")
             
             with gr.Accordion("Shared Advanced Options", open=False):  

laion10M_epoch_6_model_wo_ema.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f012c42f53f80965257c4dfcff53caeea193c5e135b5fd21f953f2e2df59406c
+size 6671914001

transfer.py

@@ -1,8 +1,11 @@
 from omegaconf import OmegaConf
 from scripts.rendertext_tool import Render_Text, load_model_from_config
 import torch
-cfg = OmegaConf.load("config_cuda_ema.yaml")
-model = load_model_from_config(cfg, "model_states.pt", verbose=True)
+
+cfg = OmegaConf.load("config_ema.yaml")
+# model = load_model_from_config(cfg, "model_states.pt", verbose=True)
+model = load_model_from_config(cfg, "mp_rank_00_model_states.pt", verbose=True)
+
 
 from pytorch_lightning.callbacks import ModelCheckpoint
 with model.ema_scope("store ema weights"):