Update app.py

app.py

@@ -12,19 +12,17 @@ import numpy as np
 import spaces 
 
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
-dtype = torch.bfloat16
 
 
 def load_models(model_path="MeissonFlow/Meissonic", 
             transformer_path="MeissonFlow/Muddit"):
     model = SymmetricTransformer2DModel.from_pretrained(
         transformer_path,
-        subfolder="1024/transformer",
-        torch_dtype=dtype
+        subfolder="1024/transformer"
     )
-    vq_model = VQModel.from_pretrained(model_path, subfolder="vqvae",torch_dtype=dtype)
-    text_encoder = CLIPTextModelWithProjection.from_pretrained(model_path, subfolder="text_encoder",torch_dtype=dtype)
-    tokenizer = CLIPTokenizer.from_pretrained(model_path, subfolder="tokenizer",torch_dtype=dtype)
+    vq_model = VQModel.from_pretrained(model_path, subfolder="vqvae")
+    text_encoder = CLIPTextModelWithProjection.from_pretrained(model_path, subfolder="text_encoder")
+    tokenizer = CLIPTokenizer.from_pretrained(model_path, subfolder="tokenizer")
     scheduler = Scheduler.from_pretrained(model_path, subfolder="scheduler")
 
     pipe = UnifiedPipeline(