jadechoghari
commited on
Commit
•
248b236
1
Parent(s):
bfffca7
Update pipeline.py
Browse files- pipeline.py +1 -6
pipeline.py
CHANGED
@@ -7,12 +7,7 @@ from huggingface_hub import hf_hub_download
|
|
7 |
from safetensors.torch import load_file
|
8 |
import os
|
9 |
from .vae import AutoencoderKL
|
10 |
-
|
11 |
-
spec = importlib.util.spec_from_file_location("mar", file_path)
|
12 |
-
mar = importlib.util.module_from_spec(spec)
|
13 |
-
sys.modules["mar"] = mar
|
14 |
-
spec.loader.exec_module(mar)
|
15 |
-
print("Successfully imported mar module!")
|
16 |
|
17 |
# inheriting from DiffusionPipeline for HF
|
18 |
class MARModel(DiffusionPipeline):
|
|
|
7 |
from safetensors.torch import load_file
|
8 |
import os
|
9 |
from .vae import AutoencoderKL
|
10 |
+
from .mar import mar
|
|
|
|
|
|
|
|
|
|
|
11 |
|
12 |
# inheriting from DiffusionPipeline for HF
|
13 |
class MARModel(DiffusionPipeline):
|