Merge branch 'release/1.9' of http://gitlab.alibaba-inc.com/Ali-MaaS/MaaS-lib into release/1.9

This commit is contained in:
mulin.lyh
2023-09-05 10:51:20 +08:00
4 changed files with 6 additions and 5 deletions

View File

@@ -40,7 +40,8 @@ class CustomCheckpointProcessor(CheckpointProcessor):
def __init__(self,
modifier_token,
modifier_token_id,
torch_type=torch.float32):
torch_type=torch.float32,
safe_serialization=False):
"""Checkpoint processor for custom diffusion.
Args:

View File

@@ -1,5 +1,5 @@
# Make sure to modify __release_datetime__ to release time when making official release.
__version__ = '1.9.0rc0'
__version__ = '1.9.0'
# default release datetime for branches under active development is set
# to be a time far-far-away-into-the-future
__release_datetime__ = '2023-09-03 00:00:00'
__release_datetime__ = '2023-09-06 00:00:00'

View File

@@ -11,7 +11,7 @@ class FaceEmotionTest(unittest.TestCase):
def setUp(self) -> None:
self.model = 'damo/cv_face-emotion'
self.img = {'img_path': 'data/test/images/face_emotion.jpg'}
self.img = 'data/test/images/face_emotion.jpg'
def pipeline_inference(self, pipeline: Pipeline, input: str):
result = pipeline(input)

View File

@@ -11,7 +11,7 @@ class HandStaticTest(unittest.TestCase):
def setUp(self) -> None:
self.model = 'damo/cv_mobileface_hand-static'
self.input = {'img_path': 'data/test/images/hand_static.jpg'}
self.input = 'data/test/images/hand_static.jpg'
def pipeline_inference(self, pipeline: Pipeline, input: str):
result = pipeline(input)