Spaces:
Running
Running
Update
Browse files
models.py
CHANGED
|
@@ -59,7 +59,7 @@ MAX_RAM_CACHE = int(float(os.environ.get('RAM_CACHE_GB', '0')) * 1e9)
|
|
| 59 |
|
| 60 |
def get_cached_model(
|
| 61 |
model_name: str,
|
| 62 |
-
) -> tuple[paligemma_bv.PaliGemmaModel, paligemma_bv.ParamsCpu]:
|
| 63 |
"""Returns model and params, using RAM cache."""
|
| 64 |
res, seq = MODELS_RES_SEQ[model_name]
|
| 65 |
model_path = gradio_helpers.get_paths()[model_name]
|
|
|
|
| 59 |
|
| 60 |
def get_cached_model(
|
| 61 |
model_name: str,
|
| 62 |
+
):# -> tuple[paligemma_bv.PaliGemmaModel, paligemma_bv.ParamsCpu]:
|
| 63 |
"""Returns model and params, using RAM cache."""
|
| 64 |
res, seq = MODELS_RES_SEQ[model_name]
|
| 65 |
model_path = gradio_helpers.get_paths()[model_name]
|