Add model:nova-d48w1024-sdxl1024

app.py

@@ -32,7 +32,7 @@ os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True"
 def parse_args():
     """Parse arguments."""
     parser = argparse.ArgumentParser(description="Serve NOVA T2I application")
-    parser.add_argument("--model", default="", help="model path")
+    parser.add_argument("--model", default="BAAI/nova-d48w1024-sdxl1024", help="model path")
     parser.add_argument("--device", type=int, default=0, help="device index")
     parser.add_argument("--precision", default="float16", help="compute precision")
     return parser.parse_args()
@@ -86,9 +86,9 @@ examples = [
 if __name__ == "__main__":
     args = parse_args()
 
-    device_type = "cuda" if torch.cuda.is_available() else "cpu"
-    device, dtype = torch.device(device_type, args.device), getattr(torch, args.precision.lower())
-    pipe = NOVAPipeline.from_pretrained(args.model, torch_dtype=dtype).to(device)
+    device = torch.device("cuda" if torch.cuda.is_available() else "cpu", args.device)
+    model_args = {"torch_dtype": getattr(torch, args.precision.lower()), "trust_remote_code": True}
+    pipe = NOVAPipeline.from_pretrained(args.model, **model_args).to(device)
 
     # Main Application.
     app = gr.Blocks(css=css, theme="origin").__enter__()

requirements.txt

@@ -1,3 +1,4 @@
+einops
 torch
 diffusers
 transformers