update app

app.py

@@ -45,7 +45,7 @@ def infer(prompt):
         config_1024 = OmegaConf.load(config_1024)
         model_config_1024 = config_1024.pop("model", OmegaConf.create())
         model_1024 = instantiate_from_config(model_config_1024)
-        # model_1024 = model_1024.cuda()
+        model_1024 = model_1024.cuda()
         model_1024 = load_model_checkpoint(model_1024, ckpt_path_1024)
         model_1024.eval()
         model = model_1024
@@ -57,7 +57,7 @@ def infer(prompt):
         config_256 = OmegaConf.load(config_256)
         model_config_256 = config_256.pop("model", OmegaConf.create())
         model_256 = instantiate_from_config(model_config_256)
-        # model_256 = model_256.cuda()
+        model_256 = model_256.cuda()
         model_256 = load_model_checkpoint(model_256, ckpt_path_256)
         model_256.eval()
         model = model_256