Update app.py

app.py

@@ -17,6 +17,7 @@ try:
         "meta-llama/Meta-Llama-3.1-8B-Instruct",
         output_hidden_states=True,
         output_attentions=True,
+        attn_implementation="eager",
         torch_dtype=dtype,
     ).to(device)
 except Exception as e:
@@ -25,7 +26,7 @@ except Exception as e:
 # Load the HiDreamImagePipeline
 try:
     pipe = HiDreamImagePipeline.from_pretrained(
-        "HiDream-ai/HiDream-I1-Dev",
+        "HiDream-ai/HiDream-I1-Fast",
         tokenizer_4=tokenizer_4,
         text_encoder_4=text_encoder_4,
         torch_dtype=dtype,
@@ -40,7 +41,7 @@ MAX_IMAGE_SIZE = 2048
 
 # Inference function with GPU access
 @spaces.GPU()
-def infer(prompt, negative_prompt="", seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=28, guidance_scale=3.5, progress=gr.Progress(track_tqdm=True)):
+def infer(prompt, negative_prompt="", seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=16, guidance_scale=3.5, progress=gr.Progress(track_tqdm=True)):
     try:
         if randomize_seed:
             seed = random.randint(0, MAX_SEED)
@@ -126,7 +127,7 @@ with gr.Blocks(css=css) as app:
                         with gr.Row():
                             steps = gr.Slider(
                                 label="Inference Steps",
-                                value=28,
+                                value=16,
                                 minimum=1,
                                 maximum=100,
                                 step=1