Texttra
/

BhoriKontext

template:diffusion-lora

Model card Files Files and versions

Texttra commited on Jul 8, 2025

Commit

a2bc943

·

verified ·

1 Parent(s): 905bc0d

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -12,7 +12,7 @@ class EndpointHandler:
         # Load base model from Hugging Face
         self.pipe = FluxKontextPipeline.from_pretrained(
             "black-forest-labs/FLUX.1-Kontext-dev",
-            torch_dtype=torch.float16,
         )
         # Debug available methods on pipeline

         # Load base model from Hugging Face
         self.pipe = FluxKontextPipeline.from_pretrained(
             "black-forest-labs/FLUX.1-Kontext-dev",
+            torch_dtype=torch.float32,
         )
         # Debug available methods on pipeline