[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
ea81279964
commit
9726f20661
|
@ -16,6 +16,7 @@
|
||||||
|
|
||||||
import torch.nn as nn
|
import torch.nn as nn
|
||||||
|
|
||||||
|
|
||||||
class ActionProjector(nn.Module):
|
class ActionProjector(nn.Module):
|
||||||
def __init__(self, in_dim, out_dim=1024):
|
def __init__(self, in_dim, out_dim=1024):
|
||||||
super().__init__()
|
super().__init__()
|
||||||
|
|
|
@ -19,6 +19,7 @@ import torch
|
||||||
from PIL import Image
|
from PIL import Image
|
||||||
from qwen_vl_utils import fetch_image
|
from qwen_vl_utils import fetch_image
|
||||||
|
|
||||||
|
|
||||||
class Qwen2VLAProcess:
|
class Qwen2VLAProcess:
|
||||||
def __init__(
|
def __init__(
|
||||||
self,
|
self,
|
||||||
|
|
Loading…
Reference in New Issue