Texttra
/

BhoriKontext

template:diffusion-lora

Model card Files Files and versions

Texttra commited on Jul 10

Commit

f4cc8cc

·

verified ·

1 Parent(s): 8db5813

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -23,7 +23,7 @@ class EndpointHandler:
         try:
             self.pipe.load_lora_weights(
                 "Texttra/BhoriKontext",
-                weight_name="Bh0r1.safetensors"
             )
             print("✅ LoRA weights loaded from Texttra/BhoriKontext/Bh0r1.safetensors.")
         except Exception as e:

         try:
             self.pipe.load_lora_weights(
                 "Texttra/BhoriKontext",
+                weight_name="Bh0r12.safetensors"
             )
             print("✅ LoRA weights loaded from Texttra/BhoriKontext/Bh0r1.safetensors.")
         except Exception as e: