We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 5c81ccb commit 62ab236Copy full SHA for 62ab236
tests/models/test_modeling_common.py
@@ -60,10 +60,6 @@ class ModelUtilsTest(unittest.TestCase):
60
def tearDown(self):
61
super().tearDown()
62
63
- import diffusers
64
-
65
- diffusers.utils.import_utils._safetensors_available = True
66
67
def test_accelerate_loading_error_message(self):
68
with self.assertRaises(ValueError) as error_context:
69
UNet2DConditionModel.from_pretrained("hf-internal-testing/stable-diffusion-broken", subfolder="unet")
0 commit comments