mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-12-22 04:20:49 +08:00
fix log
This commit is contained in:
parent
e9e1d2f0e8
commit
49561788cf
@ -797,7 +797,7 @@ class ModelPatcher:
|
|||||||
tensors to memory-mapped format instead.
|
tensors to memory-mapped format instead.
|
||||||
"""
|
"""
|
||||||
import pdb; pdb.set_trace()
|
import pdb; pdb.set_trace()
|
||||||
logging.info(f"model {self.model.__class__.__name__} is calling to_mmap method")
|
logging.info(f"model {self.__class__.__name__} is calling to_mmap method")
|
||||||
def convert_fn(t):
|
def convert_fn(t):
|
||||||
if isinstance(t, torch.Tensor) and not isinstance(t, torch.nn.Parameter):
|
if isinstance(t, torch.Tensor) and not isinstance(t, torch.nn.Parameter):
|
||||||
return to_mmap(t)
|
return to_mmap(t)
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user