add tests

This commit is contained in:
yuze.zyz
2025-01-26 18:50:28 +08:00
parent cf6aa132de
commit 565e45370e

View File

@@ -150,6 +150,11 @@ class HFUtilTest(unittest.TestCase):
else:
self.assertTrue(False)
from modelscope import StableDiffusionPipeline
pipe = StableDiffusionPipeline.from_pretrained(
'AI-ModelScope/stable-diffusion-v1-5')
self.assertTrue(pipe is not None)
def test_patch_peft(self):
with patch_context():
from transformers import AutoModelForCausalLM