Texttra
/

BhoriKontext

template:diffusion-lora

Model card Files Files and versions

Texttra commited on Jul 6, 2025

Commit

de4914e

·

verified ·

1 Parent(s): ec2c1d4

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -12,7 +12,7 @@ class EndpointHandler:
         # Load Flux Kontext model
         self.pipe = FluxKontextPipeline.from_pretrained(
             "black-forest-labs/FLUX.1-Kontext-dev",
-            torch_dtype=torch.float16,
         )
         self.pipe.to("cuda" if torch.cuda.is_available() else "cpu")
         print("✅ Model ready.")

         # Load Flux Kontext model
         self.pipe = FluxKontextPipeline.from_pretrained(
             "black-forest-labs/FLUX.1-Kontext-dev",
+            torch_dtype=torch.float32,
         )
         self.pipe.to("cuda" if torch.cuda.is_available() else "cpu")
         print("✅ Model ready.")