mirror of
https://git.datalinker.icu/comfyanonymous/ComfyUI
synced 2026-08-16 00:16:39 +08:00
Typo.
This commit is contained in:
parent
4d7019555d
commit
f2c38dd9dc
@ -287,7 +287,7 @@ class Llama2_(nn.Module):
|
|||||||
x *= self.config.hidden_size ** 0.5
|
x *= self.config.hidden_size ** 0.5
|
||||||
|
|
||||||
if position_ids is None:
|
if position_ids is None:
|
||||||
position_ids = torch.arange(0, x.shape[1], device=xdevice).unsqueeze(0)
|
position_ids = torch.arange(0, x.shape[1], device=x.device).unsqueeze(0)
|
||||||
|
|
||||||
freqs_cis = precompute_freqs_cis(self.config.head_dim,
|
freqs_cis = precompute_freqs_cis(self.config.head_dim,
|
||||||
position_ids,
|
position_ids,
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user