wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-25 22:00:19 +08:00
Code Issues Actions 16 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 import math
class EPS: class EPS:
sigma_data: float = None sigma_data: float
def calculate_input(self, sigma, noise): def calculate_input(self, sigma, noise):
sigma = sigma.view(sigma.shape[:1] + (1,) * (noise.ndim - 1)) 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() backup_embeds = self.transformer.get_input_embeddings()
device = backup_embeds.weight.device device = backup_embeds.weight.device
tokens = self.set_up_textual_embeddings(tokens, backup_embeds) 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 attention_mask = None
if self.enable_attention_masks: if self.enable_attention_masks: