修改参数

app.py

@@ -24,8 +24,9 @@ base_path = "black-forest-labs/FLUX.1-dev"
 lora_base_path = "./models"
 
 
-pipe = FluxPipeline.from_pretrained(base_path, torch_dtype=torch.bfloat16)
-transformer = FluxTransformer2DModel.from_pretrained(base_path, subfolder="transformer", torch_dtype=torch.bfloat16)
+# pipe = FluxPipeline.from_pretrained(base_path, torch_dtype=torch.bfloat16)
+pipe = FluxPipeline.from_pretrained(base_path, torch_dtype=torch.bfloat8)
+transformer = FluxTransformer2DModel.from_pretrained(base_path, subfolder="transformer", torch_dtype=torch.bfloat8)
 pipe.transformer = transformer
 pipe.to("cuda")