Spaces:
Paused
Paused
alessandro trinca tornidor
commited on
Commit
·
3db919f
1
Parent(s):
32cf4a3
feat: fix wrong typing hints that caused exception on app.py start because of gradio
Browse files
lisa_on_cuda/utils/app_helpers.py
CHANGED
|
@@ -293,7 +293,7 @@ def prepare_model_vision_tower(_model, args_to_parse, torch_dtype, internal_logg
|
|
| 293 |
|
| 294 |
|
| 295 |
def get_inference_model_by_args(
|
| 296 |
-
args_to_parse, internal_logger0: logging = None, inference_decorator: Callable = None, device_map="auto", device="cuda"
|
| 297 |
):
|
| 298 |
"""Load model and inference function with arguments. Compatible with ZeroGPU (spaces 0.30.2)
|
| 299 |
|
|
@@ -317,7 +317,7 @@ def get_inference_model_by_args(
|
|
| 317 |
def inference(
|
| 318 |
input_str: str,
|
| 319 |
input_image: str | Path | np.ndarray,
|
| 320 |
-
internal_logger: logging = None,
|
| 321 |
embedding_key: str = None
|
| 322 |
):
|
| 323 |
if internal_logger is None:
|
|
@@ -471,7 +471,7 @@ def get_gradio_interface(
|
|
| 471 |
description=constants.description,
|
| 472 |
article=article_and_demo_parameters,
|
| 473 |
examples=constants.examples,
|
| 474 |
-
|
| 475 |
)
|
| 476 |
|
| 477 |
|
|
|
|
| 293 |
|
| 294 |
|
| 295 |
def get_inference_model_by_args(
|
| 296 |
+
args_to_parse, internal_logger0: logging.Logger = None, inference_decorator: Callable = None, device_map="auto", device="cuda"
|
| 297 |
):
|
| 298 |
"""Load model and inference function with arguments. Compatible with ZeroGPU (spaces 0.30.2)
|
| 299 |
|
|
|
|
| 317 |
def inference(
|
| 318 |
input_str: str,
|
| 319 |
input_image: str | Path | np.ndarray,
|
| 320 |
+
internal_logger: logging.Logger = None,
|
| 321 |
embedding_key: str = None
|
| 322 |
):
|
| 323 |
if internal_logger is None:
|
|
|
|
| 471 |
description=constants.description,
|
| 472 |
article=article_and_demo_parameters,
|
| 473 |
examples=constants.examples,
|
| 474 |
+
# flagging_mode="auto"
|
| 475 |
)
|
| 476 |
|
| 477 |
|