diff --git a/.gitignore b/.gitignore index 10fc461..cae4225 100644 --- a/.gitignore +++ b/.gitignore @@ -1,4 +1,5 @@ # Byte-compiled / optimized / DLL files +model __pycache__/ *.py[cod] *$py.class diff --git a/scripts/example.py b/scripts/example.py index 8bb08a3..ba66b5d 100644 --- a/scripts/example.py +++ b/scripts/example.py @@ -333,7 +333,7 @@ def prepare_prompt(*, prompt: str, prompt_assets: dict, views: list[str]): } # add mask token["mask"] = { - view: np.ones((n_objs_prompt[view],), dtype=np.bool) + view: np.ones((n_objs_prompt[view],), dtype=bool) for view in views } n_objs_to_pad = { @@ -353,7 +353,7 @@ def prepare_prompt(*, prompt: str, prompt_assets: dict, views: list[str]): for view in views }, "mask": { - view: np.zeros((n_objs_to_pad[view]), dtype=np.bool) + view: np.zeros((n_objs_to_pad[view]), dtype=bool) for view in views }, }