mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-09-11 03:58:22 +00:00
Automatically use fp8 for diffusion model weights if:
Checkpoint contains weights in fp8. There isn't enough memory to load the diffusion model in GPU vram.
This commit is contained in:
@@ -94,6 +94,7 @@ class BaseModel(torch.nn.Module):
|
||||
self.concat_keys = ()
|
||||
logging.info("model_type {}".format(model_type.name))
|
||||
logging.debug("adm {}".format(self.adm_channels))
|
||||
logging.info("model weight dtype {}, manual cast: {}".format(self.get_dtype(), self.manual_cast_dtype))
|
||||
self.memory_usage_factor = model_config.memory_usage_factor
|
||||
|
||||
def apply_model(self, x, t, c_concat=None, c_crossattn=None, control=None, transformer_options={}, **kwargs):
|
||||
|
Reference in New Issue
Block a user