Spaces:

yonnel
/

text-to-3d_flux_trellis

Runtime error

lionelgarnier commited on Feb 18

Commit

0d131d4

1 Parent(s): e6ef8b4

back to flux schnell

Files changed (1) hide show

app.py CHANGED Viewed

@@ -27,8 +27,8 @@ def get_image_gen_pipeline():
             device = "cuda" if torch.cuda.is_available() else "cpu"
             dtype = torch.bfloat16
             _image_gen_pipeline = DiffusionPipeline.from_pretrained(
-                # "black-forest-labs/FLUX.1-schnell",
-                "black-forest-labs/FLUX.1-dev",
                 torch_dtype=dtype,
             ).to(device)
             _image_gen_pipeline.enable_model_cpu_offload()
@@ -175,6 +175,7 @@ def preload_models():
         dtype = torch.bfloat16
         _image_gen_pipeline = DiffusionPipeline.from_pretrained(
             "black-forest-labs/FLUX.1-schnell",
             torch_dtype=dtype
         ).to(device)

             device = "cuda" if torch.cuda.is_available() else "cpu"
             dtype = torch.bfloat16
             _image_gen_pipeline = DiffusionPipeline.from_pretrained(
+                "black-forest-labs/FLUX.1-schnell",
+                # "black-forest-labs/FLUX.1-dev",
                 torch_dtype=dtype,
             ).to(device)
             _image_gen_pipeline.enable_model_cpu_offload()
         dtype = torch.bfloat16
         _image_gen_pipeline = DiffusionPipeline.from_pretrained(
             "black-forest-labs/FLUX.1-schnell",
+            # "black-forest-labs/FLUX.1-dev",
             torch_dtype=dtype
         ).to(device)