Huong commited on
Commit
001ffb6
·
1 Parent(s): 791cc99

updated app.py with tested downloading and converting of model

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -28,12 +28,10 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
28
  torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
29
 
30
  # Configuration for model download and conversion
31
- OLMOASR_REPO = "allenai/OLMoASR" # Temporary model link as requested
32
- CHECKPOINT_FILENAME = (
33
- "OLMoASR-medium.en.pt" # Adjust based on actual filename in the repo
34
- )
35
  LOCAL_CHECKPOINT_DIR = "checkpoints"
36
- HF_MODEL_DIR = "checkpoints/medium_hf_converted"
37
 
38
 
39
  def ensure_checkpoint_dir():
 
28
  torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
29
 
30
  # Configuration for model download and conversion
31
+ OLMOASR_REPO = "olmoasr/OLMoASR-small.en" # Temporary model link as requested
32
+ CHECKPOINT_FILENAME = "latesttrain_00524288_small_ddp-train_grad-acc_fp16_non_ddp_inf.pt" # Adjust based on actual filename in the repo
 
 
33
  LOCAL_CHECKPOINT_DIR = "checkpoints"
34
+ HF_MODEL_DIR = "checkpoints/small_hf_converted"
35
 
36
 
37
  def ensure_checkpoint_dir():