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:
|
if RMSNorm is None:
|
||||||
class RMSNorm(torch.nn.Module):
|
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__(
|
def __init__(
|
||||||
self,
|
self,
|
||||||
normalized_shape,
|
normalized_shape,
|
||||||
|
@ -1,5 +1,3 @@
|
|||||||
from comfy import sd1_clip
|
|
||||||
import os
|
|
||||||
|
|
||||||
|
|
||||||
def model_options_long_clip(sd, tokenizer_data, model_options):
|
def model_options_long_clip(sd, tokenizer_data, model_options):
|
||||||
|
Loading…
Reference in New Issue
Block a user