Commit
•
f2191a0
1
Parent(s):
1822fe2
Update pipeline_mar.py
Browse files- pipeline_mar.py +2 -2
pipeline_mar.py
CHANGED
|
@@ -4,8 +4,8 @@ import numpy as np
|
|
| 4 |
from huggingface_hub import hf_hub_download
|
| 5 |
from safetensors.torch import load_file
|
| 6 |
import os
|
| 7 |
-
from
|
| 8 |
-
|
| 9 |
|
| 10 |
# inheriting from DiffusionPipeline for HF
|
| 11 |
class MARModel(DiffusionPipeline):
|
|
|
|
| 4 |
from huggingface_hub import hf_hub_download
|
| 5 |
from safetensors.torch import load_file
|
| 6 |
import os
|
| 7 |
+
from .vae import AutoencoderKL
|
| 8 |
+
import .mar
|
| 9 |
|
| 10 |
# inheriting from DiffusionPipeline for HF
|
| 11 |
class MARModel(DiffusionPipeline):
|