AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

[Tests] Mark the ncsnpp model tests as slow (#575)

Browse Source 
* [Tests] Mark the ncsnpp model tests as slow

* style
This commit is contained in:
Anton Lozhkov
2022-09-19 17:20:58 +02:00
committed by GitHub
parent 0a2c42f3e2
commit 9727cda678
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_models_unet.py
View File

@@ -19,7 +19,7 @@ import unittest
import torch
from diffusers import UNet2DModel
from diffusers.testing_utils import floats_tensor, torch_device
from diffusers.testing_utils import floats_tensor, slow, torch_device
from .test_modeling_common import ModelTesterMixin
@@ -231,6 +231,7 @@ class NCSNppModelTests(ModelTesterMixin, unittest.TestCase):
inputs_dict = self.dummy_input
return init_dict, inputs_dict
@slow
def test_from_pretrained_hub(self):
model, loading_info = UNet2DModel.from_pretrained("google/ncsnpp-celebahq-256", output_loading_info=True)
self.assertIsNotNone(model)
@@ -244,6 +245,7 @@ class NCSNppModelTests(ModelTesterMixin, unittest.TestCase):
assert image is not None, "Make sure output is not None"
@slow
def test_output_pretrained_ve_mid(self):
model = UNet2DModel.from_pretrained("google/ncsnpp-celebahq-256")
model.to(torch_device)