Update processor.py
Browse files- processor.py +1 -2
processor.py
CHANGED
@@ -2,7 +2,6 @@ import subprocess
|
|
2 |
import importlib
|
3 |
import sys
|
4 |
import logging
|
5 |
-
from transformers import ProcessorMixin
|
6 |
import torch
|
7 |
import numpy as np
|
8 |
from rembg import remove, new_session
|
@@ -17,7 +16,7 @@ import kiui
|
|
17 |
# and might be causing issues if run repeatedly
|
18 |
|
19 |
|
20 |
-
class LRMImageProcessor(
|
21 |
def __init__(self, source_size=512, *args, **kwargs):
|
22 |
print("add super")
|
23 |
super().__init__(*args, **kwargs)
|
|
|
2 |
import importlib
|
3 |
import sys
|
4 |
import logging
|
|
|
5 |
import torch
|
6 |
import numpy as np
|
7 |
from rembg import remove, new_session
|
|
|
16 |
# and might be causing issues if run repeatedly
|
17 |
|
18 |
|
19 |
+
class LRMImageProcessor():
|
20 |
def __init__(self, source_size=512, *args, **kwargs):
|
21 |
print("add super")
|
22 |
super().__init__(*args, **kwargs)
|