modify

README.md

@@ -6,7 +6,7 @@ colorTo: red
 sdk: gradio
 sdk_version: 5.0.1
 app_file: app.py
-pinned: false
+pinned: true
 license: apache-2.0
 short_description: Diffusion Model Compression
 ---

app.py

@@ -9,10 +9,7 @@ import torch
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model_repo_id = "stabilityai/stable-diffusion-xl-base-1.0"  # Replace to the model you would like to use
 
-if torch.cuda.is_available():
-    torch_dtype = torch.float16
-else:
-    torch_dtype = torch.float32
+torch_dtype = torch.bfloat16
 
 pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
 pipe = pipe.to(device)

requirements.txt

@@ -1,6 +1,6 @@
-accelerate
-diffusers
+diffusers==0.31.0
 invisible_watermark
 torch
-transformers
-xformers
+sentencepiece
+transformers==4.45.2
+accelerate==0.33.0