diff --git a/comfy/model_sampling.py b/comfy/model_sampling.py index 7bdd3636d..116156728 100644 --- a/comfy/model_sampling.py +++ b/comfy/model_sampling.py @@ -3,7 +3,7 @@ from .ldm.modules.diffusionmodules.util import make_beta_schedule import math class EPS: - sigma_data: float = None + sigma_data: float def calculate_input(self, sigma, noise): sigma = sigma.view(sigma.shape[:1] + (1,) * (noise.ndim - 1)) diff --git a/comfy/sd1_clip.py b/comfy/sd1_clip.py index 68c3b69ad..2ac014376 100644 --- a/comfy/sd1_clip.py +++ b/comfy/sd1_clip.py @@ -171,7 +171,7 @@ class SDClipModel(torch.nn.Module, ClipTokenWeightEncoder): backup_embeds = self.transformer.get_input_embeddings() device = backup_embeds.weight.device tokens = self.set_up_textual_embeddings(tokens, backup_embeds) - tokens = torch.LongTensor(tokens).to(device) + tokens = torch.Tensor(tokens, dtype=torch.long).to(device) attention_mask = None if self.enable_attention_masks: