Commit
•
1ce37cf
1
Parent(s):
225a84f
Update processor.py
Browse files- processor.py +2 -1
processor.py
CHANGED
|
@@ -2,6 +2,7 @@ import subprocess
|
|
| 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,7 +17,7 @@ import kiui
|
|
| 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)
|
|
|
|
| 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 |
# and might be causing issues if run repeatedly
|
| 18 |
|
| 19 |
|
| 20 |
+
class LRMImageProcessor(ProcessorMixin):
|
| 21 |
def __init__(self, source_size=512, *args, **kwargs):
|
| 22 |
print("add super")
|
| 23 |
super().__init__(*args, **kwargs)
|