Convert input_ids to float32

app.py

@@ -27,7 +27,9 @@ scheduler = LMSDiscreteScheduler(beta_start=beta_start, beta_end=beta_end, beta_
 def text_enc(prompts, maxlen=None):
     if maxlen is None: maxlen = tokenizer.model_max_length
     inp = tokenizer(prompts, padding="max_length", max_length=maxlen, truncation=True, return_tensors="pt")
-    return text_encoder(inp.input_ids.long())[0]
+    input_ids = inp.input_ids.long()  # Convert input_ids to Long data type
+    input_ids = input_ids.float()  # Convert input_ids to float32
+    return text_encoder(input_ids)[0]
 
 def do_both(prompts):
     def mk_img(t):