wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-11 23:00:51 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

Change tokens to tensor of type long

Browse Source
This commit is contained in:
Max Tretikov 2024-06-13 15:25:15 -06:00
parent 63636c3355
commit d7f7d81b8a
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/model_sampling.py
View File

@ -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))

2
comfy/sd1_clip.py
View File

@ -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: