Texttra
/

Cityscape_Studio

template:diffusion-lora

Model card Files Files and versions

Texttra commited on Apr 7, 2025

Commit

1ccb9bc

·

verified ·

1 Parent(s): 4f82630

Update handler.py

Files changed (1) hide show

handler.py +0 -1

handler.py CHANGED Viewed

@@ -11,7 +11,6 @@ class EndpointHandler:
         self.pipe = DiffusionPipeline.from_pretrained(
             "black-forest-labs/FLUX.1-dev",
             torch_dtype=torch.float16,
-            variant="fp16",
             use_auth_token=True
         )

         self.pipe = DiffusionPipeline.from_pretrained(
             "black-forest-labs/FLUX.1-dev",
             torch_dtype=torch.float16,
             use_auth_token=True
         )