model: Support Janus-pro (#3203)

2025-03-12 11:02:11 -07:00
parent 6f43a9b9f4
commit 01090e8ac3
13 changed files with 2957 additions and 15 deletions
@@ -6,7 +6,7 @@ from typing import Optional, Tuple
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
-from einops import rearrange, repeat
+from einops import rearrange

 from sglang.srt.distributed import parallel_state
 from sglang.srt.distributed import utils as dist_utils