yyk19 commited on
Commit
98333ba
1 Parent(s): ecc45b8

delete a checkpoint.

Browse files
Files changed (2) hide show
  1. app.py +4 -4
  2. laion10M_epoch_5_model_wo_ema.ckpt +0 -3
app.py CHANGED
@@ -84,9 +84,9 @@ def load_ckpt(model_ckpt = "LAION-Glyph-10M-Epoch-5"):
84
 
85
  # if model_ckpt == "LAION-Glyph-1M":
86
  # model = load_model_ckpt(model, "laion1M_model_wo_ema.ckpt")
87
- if model_ckpt == "LAION-Glyph-10M-Epoch-5":
88
- model = load_model_ckpt(model, "laion10M_epoch_5_model_wo_ema.ckpt")
89
- elif model_ckpt == "LAION-Glyph-10M-Epoch-6":
90
  model = load_model_ckpt(model, "laion10M_epoch_6_model_wo_ema.ckpt")
91
  elif model_ckpt == "TextCaps-5K-Epoch-10":
92
  model = load_model_ckpt(model, "textcaps5K_epoch_10_model_wo_ema.ckpt")
@@ -161,7 +161,7 @@ with block:
161
  with gr.Row():
162
  # model_ckpt = gr.inputs.Dropdown(["LAION-Glyph-10M", "Textcaps5K-10"], label="Checkpoint", default = "LAION-Glyph-10M")
163
  # 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")
164
- model_ckpt = gr.inputs.Dropdown(["LAION-Glyph-10M-Epoch-6", "LAION-Glyph-10M-Epoch-5", "TextCaps-5K-Epoch-10", "TextCaps-5K-Epoch-20", "TextCaps-5K-Epoch-40"], label="Checkpoint", default = "LAION-Glyph-10M-Epoch-6")
165
  # load_button = gr.Button(value = "Load Checkpoint")
166
 
167
  with gr.Accordion("Shared Advanced Options", open=False):
 
84
 
85
  # if model_ckpt == "LAION-Glyph-1M":
86
  # model = load_model_ckpt(model, "laion1M_model_wo_ema.ckpt")
87
+ # if model_ckpt == "LAION-Glyph-10M-Epoch-5":
88
+ # model = load_model_ckpt(model, "laion10M_epoch_5_model_wo_ema.ckpt")
89
+ if model_ckpt == "LAION-Glyph-10M-Epoch-6":
90
  model = load_model_ckpt(model, "laion10M_epoch_6_model_wo_ema.ckpt")
91
  elif model_ckpt == "TextCaps-5K-Epoch-10":
92
  model = load_model_ckpt(model, "textcaps5K_epoch_10_model_wo_ema.ckpt")
 
161
  with gr.Row():
162
  # model_ckpt = gr.inputs.Dropdown(["LAION-Glyph-10M", "Textcaps5K-10"], label="Checkpoint", default = "LAION-Glyph-10M")
163
  # 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")
164
+ model_ckpt = gr.inputs.Dropdown(["LAION-Glyph-10M-Epoch-6", "TextCaps-5K-Epoch-10", "TextCaps-5K-Epoch-20", "TextCaps-5K-Epoch-40"], label="Checkpoint", default = "LAION-Glyph-10M-Epoch-6")
165
  # load_button = gr.Button(value = "Load Checkpoint")
166
 
167
  with gr.Accordion("Shared Advanced Options", open=False):
laion10M_epoch_5_model_wo_ema.ckpt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:c7ae9f29a41152a85bc5001811f80f701fb8b845b44526483497fdd6f4946e4b
3
- size 6671914001