theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-19 06:48:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Artiprocher
2025-11-20 17:44:00 +08:00
parent eeb55a0ce6
commit 96daa30bcc
7 changed files with 72 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
diffsynth/core/vram/disk_map.py
View File

@@ -60,9 +60,10 @@ class DiskMap:
if self.rename_dict is not None: name = self.rename_dict[name]
file_id = self.name_map[name]
param = self.files[file_id].get_tensor(name)
if self.torch_dtype is not None:
if self.torch_dtype is not None and isinstance(param, torch.Tensor):
param = param.to(self.torch_dtype)
self.num_params += param.numel()
if isinstance(param, torch.Tensor):
self.num_params += param.numel()
if self.num_params > self.buffer_size:
self.flush_files()
return param