Texttra
/

BhoriKontext

template:diffusion-lora

Model card Files Files and versions

Texttra commited on Jul 9

Commit

ad5c32f

·

verified ·

1 Parent(s): 2f69dee

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -13,7 +13,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.bfloat16,
         )
         # Debug available methods on pipeline