ACE-Plus

Running on Zero

chaojiemao commited on Dec 7, 2024

Commit

38484eb

verified ·

1 Parent(s): 3ea19cb

Update ace_inference.py

Files changed (1) hide show

ace_inference.py CHANGED Viewed

@@ -84,7 +84,7 @@ class TextEmbedding(nn.Module):
         super().__init__()
         self.pos = nn.Parameter(data=torch.zeros(embedding_shape))
-class ACEFluxLCInference(DiffusionInference):
     def __init__(self, logger=None):
         if logger is None:
             logger = get_logger(name='scepter')

         super().__init__()
         self.pos = nn.Parameter(data=torch.zeros(embedding_shape))
+class ACEInference(DiffusionInference):
     def __init__(self, logger=None):
         if logger is None:
             logger = get_logger(name='scepter')