Update __init__.py
Browse files- __init__.py +2 -0
__init__.py
CHANGED
|
@@ -1,4 +1,6 @@
|
|
|
|
|
| 1 |
from .modeling import LRMGenerator, LRMGeneratorConfig
|
| 2 |
from .processor import LRMImageProcessor
|
| 3 |
|
|
|
|
| 4 |
AutoProcessor.register(LRMImageProcessor)
|
|
|
|
| 1 |
+
from transformers import AutoProcessor
|
| 2 |
from .modeling import LRMGenerator, LRMGeneratorConfig
|
| 3 |
from .processor import LRMImageProcessor
|
| 4 |
|
| 5 |
+
# we add the preprocessor
|
| 6 |
AutoProcessor.register(LRMImageProcessor)
|