AiArtLab
/

sdxs-1b

Model card Files Files and versions

recoilme commited on Mar 30

Commit

32a3591

·

1 Parent(s): 5b9bbdb

1.6b

Files changed (1) hide show

train.py +1 -1

train.py CHANGED Viewed

@@ -171,7 +171,7 @@ def load_text_encoder():
         ).to(device).eval()
 @torch.no_grad()
-def encode_texts(text, max_length=max_length)::
     device = self.device
     dtype = self.unet.dtype
     if text is None: text = ""

         ).to(device).eval()
 @torch.no_grad()
+def encode_texts(text, max_length=max_length):
     device = self.device
     dtype = self.unet.dtype
     if text is None: text = ""