init

app.py

@@ -33,8 +33,7 @@ if not path.exists(cache_path):
 pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev")
 pipe.load_lora_weights(hf_hub_download("RED-AIGC/TDD", "TDD-FLUX.1-dev-lora-beta.safetensors"))
 pipe.fuse_lora(lora_scale=0.125)
-pipe.to(dtype=torch.float16)
-pipe.to("cuda")
+pipe.to(device="cuda", dtype=torch.bfloat16)
 
 css = """
 h1 {
@@ -50,8 +49,15 @@ with gr.Blocks(css=css) as demo:
     gr.Markdown(
         """
         # FLUX.1-dev(beta) distilled by ✨Target-Driven Distillation✨
-        Compared to Hyper-FLUX, the beta version of TDD has its parameters reduced by half(600MB), resulting in more realistic details. Due to the limitations of the machine configuration, there are still many imperfections in the test version. The official version is still being optimized and is expected to be released after the National Day holiday. TDD is also available for distilling video generation models. This space presents TDD-distilled [FLUX.1-dev](https://huggingface.co/black-forest-labs/FLUX.1-dev).
+        
+        Compared to Hyper-FLUX, the beta version of TDD has its parameters reduced by half(600MB), resulting in more realistic details. 
+        
+        Due to limitations in machine resources, there are still many imperfections in the beta version. The official version is still being optimized and is expected to be released after the National Day holiday. 
+        
+        Besides, TDD is also available for distilling video generation models. This space presents TDD-distilled [FLUX.1-dev](https://huggingface.co/black-forest-labs/FLUX.1-dev).
+        
         [**Project Page**](https://redaigc.github.io/TDD/) **|** [**Paper**](https://arxiv.org/abs/2409.01347) **|** [**Code**](https://github.com/RedAIGC/Target-Driven-Distillation) **|** [**Model**](https://huggingface.co/RED-AIGC/TDD) **|** [🤗 **TDD-SDXL Demo**](https://huggingface.co/spaces/RED-AIGC/TDD) **|** [🤗 **TDD-SVD Demo**](https://huggingface.co/spaces/RED-AIGC/SVD-TDD)
+        
         The codes of this space are built on [Hyper-FLUX](https://huggingface.co/spaces/ByteDance/Hyper-FLUX-8Steps-LoRA) and we acknowledge their contribution.
         """
     )