2511

test.ipynb

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a9f0437fc7ce18a2a9b98d23d5c6ada05691720b2022b43fb6572c896e99e948
-size 1598717
+oid sha256:a4705bdf48f7ee0c40ef75e30cf3fc1c52aecf1dc5438a2ad673a038ac6d3cf2
+size 1598727

train.py

@@ -132,6 +132,21 @@ def encode_texts(texts, max_length=150):
         pass
 
     with torch.no_grad():
+        if isinstance(texts, str):
+            texts = [texts]
+
+        for i, prompt_item in enumerate(texts):
+            messages = [
+                {"role": "user", "content": prompt_item},
+            ]
+            prompt_item = tokenizer.apply_chat_template(
+                messages,
+                tokenize=False,
+                add_generation_prompt=True,
+                #enable_thinking=True,
+            )
+            texts[i] = prompt_item
+            
         toks = tokenizer(
             texts,
             return_tensors="pt",

unet/config.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:061de3469380a2fd1f271252c1e66217c5118e3d346a98b001e80051683a116b
-size 1883
+oid sha256:9437f6d40639f1ad6d95d6586009d7b1bf0c8e99959d29da5c9c9645cae39ea3
+size 1899

unet/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5faa0fed4a1fb247ccdeef15253c21b27d2af2f537004b5f698dd06b357827ca
-size 6205958296
+oid sha256:ec4b1086b68631ce848c26ebea4688807df497bcbd2b099bf521a5dc64845645
+size 3103078992