mirror of
https://github.com/modelscope/modelscope.git
synced 2025-12-16 16:27:45 +01:00
add tests
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user