Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-07-13 22:08:46 +02:00
Code Issues Packages Projects Releases Wiki Activity

merge github pull #519

Browse Source 
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/13900895
* merge github pull #519

* fix bug

* fix case issue

* merge github #519
This commit is contained in:
mulin.lyh
2023-09-05 10:42:43 +08:00
parent bb63a8d465
commit 797dda70bb
3 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
modelscope/trainers/multi_modal/custom_diffusion/custom_diffusion_trainer.py
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:

2
tests/pipelines/test_face_emotion.py
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)

2
tests/pipelines/test_hand_static.py
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)