diff --git a/comfy/ldm/chroma_radiance/model.py b/comfy/ldm/chroma_radiance/model.py index c077396cb..70d173889 100644 --- a/comfy/ldm/chroma_radiance/model.py +++ b/comfy/ldm/chroma_radiance/model.py @@ -284,7 +284,7 @@ class ChromaRadiance(Chroma): # non zero during training to prevent 0 div eps = 0.0 return (noisy - predicted) / (timesteps.view(-1,1,1,1) + eps) - + def _forward( self, x: Tensor, diff --git a/comfy/model_detection.py b/comfy/model_detection.py index 13fa1fa5e..19e6aa954 100644 --- a/comfy/model_detection.py +++ b/comfy/model_detection.py @@ -257,7 +257,7 @@ def detect_unet_config(state_dict, key_prefix, metadata=None): dit_config["nerf_tile_size"] = 512 dit_config["nerf_final_head_type"] = "conv" if f"{key_prefix}nerf_final_layer_conv.norm.scale" in state_dict_keys else "linear" dit_config["nerf_embedder_dtype"] = torch.float32 - if f"__x0__" in state_dict_keys: # x0 pred + if "__x0__" in state_dict_keys: # x0 pred dit_config["use_x0"] = True else: dit_config["guidance_embed"] = "{}guidance_in.in_layer.weight".format(key_prefix) in state_dict_keys