mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-04-19 10:53:29 +00:00
Cleanup.
This commit is contained in:
parent
3e8155f7a3
commit
6fc5dbd52a
@ -27,17 +27,6 @@ def rms_norm(x, weight=None, eps=1e-6):
|
||||
|
||||
if RMSNorm is None:
|
||||
class RMSNorm(torch.nn.Module):
|
||||
def __init__(
|
||||
self, dim: int, elementwise_affine: bool = False, eps: float = 1e-6, device=None, dtype=None, **kwargs
|
||||
):
|
||||
super().__init__()
|
||||
self.eps = eps
|
||||
self.learnable_scale = elementwise_affine
|
||||
if self.learnable_scale:
|
||||
self.weight = torch.nn.Parameter(torch.empty(dim, device=device, dtype=dtype))
|
||||
else:
|
||||
self.register_parameter("weight", None)
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
normalized_shape,
|
||||
|
@ -1,5 +1,3 @@
|
||||
from comfy import sd1_clip
|
||||
import os
|
||||
|
||||
|
||||
def model_options_long_clip(sd, tokenizer_data, model_options):
|
||||
|
Loading…
Reference in New Issue
Block a user