mirror of
https://github.com/brycedrennan/imaginAIry
synced 2024-11-19 03:25:41 +00:00
967eb76365
- simpler logging suppression for `transformers` library - suppress logging noise for running tests - get test running for all samplers on mps and cuda platforms - refactor safety model env variable to allow classification
13 lines
319 B
Python
13 lines
319 B
Python
def imagine_cmd(*args, **kwargs):
|
|
from .suppress_logs import suppress_annoying_logs_and_warnings # noqa
|
|
|
|
suppress_annoying_logs_and_warnings()
|
|
|
|
from imaginairy.cmds import imagine_cmd as imagine_cmd_orig # noqa
|
|
|
|
imagine_cmd_orig(*args, **kwargs)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
imagine_cmd() # noqa
|