Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-25 04:29:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix text_to_video_synthesis_model device (#751)

Browse Source 
Co-authored-by: slin000111 <zhaoshilin2015@sina.con>
This commit is contained in:
slin000111
2024-02-23 17:05:15 +08:00
committed by GitHub
parent 2df76a9304
commit 8a8d273ec3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modelscope/models/multi_modal/video_synthesis/text_to_video_synthesis_model.py
View File

@@ -58,7 +58,7 @@ class TextToVideoSynthesis(Model):
`True`.
"""
super().__init__(model_dir=model_dir, *args, **kwargs)
self.device = torch.device('cuda') if torch.cuda.is_available() \
self.device = torch.device(kwargs.get('device', 'cuda')) if torch.cuda.is_available() \
else torch.device('cpu')
self.config = Config.from_file(
osp.join(model_dir, ModelFile.CONFIGURATION))