fix

app.py

@@ -1,6 +1,7 @@
 import logging
 import gradio as gr
 import torch
+import numpy as np
 from transformers import (
     MODEL_FOR_MASKED_LM_MAPPING,
 )
@@ -67,6 +68,10 @@ def main():
             pipeline.simplex_value = simplex_value
             pipeline.top_p = top_p
             pipeline.temperature = temperature
+            
+            # Ensure timesteps are properly set as a sequence
+            pipeline.scheduler.timesteps = torch.arange(0, diffusion_steps).flip(0)
+            
             # tulu chat template
             inputs = "<|user|>\n" + inputs + "<|assistant|>\n"