Huong
commited on
Commit
·
001ffb6
1
Parent(s):
791cc99
updated app.py with tested downloading and converting of model
Browse files
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 = "
|
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/
|
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():
|