Skip to content

Conversation

@anton-l
Copy link
Member

@anton-l anton-l commented Sep 20, 2022

Checks whether the models and model cards get saved after a (successful) training run, nothing more!

To run locally, run python -m pytest -s -v ./examples from the root dir.

Comment on lines 71 to 77
def test_train_unconditional(self):
stream_handler = logging.StreamHandler(sys.stdout)
logger.addHandler(stream_handler)

with tempfile.TemporaryDirectory() as tmpdir:
test_args = f"""
examples/unconditional_image_generation/train_unconditional.py
Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This test will pass after #570 is merged

@HuggingFaceDocBuilderDev
Copy link

HuggingFaceDocBuilderDev commented Sep 20, 2022

The documentation is not available anymore as the PR was closed or merged.

Copy link
Contributor

@patil-suraj patil-suraj left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks good! Thank you for adding these tests !

@patrickvonplaten
Copy link
Contributor

Very nice idea! Also good to test something!

@anton-l anton-l merged commit 4f1c989 into main Sep 21, 2022
@anton-l anton-l deleted the test-examples branch September 21, 2022 11:37
PhaneeshB pushed a commit to nod-ai/diffusers that referenced this pull request Mar 1, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants