Commit
•
4bd6d0e
1
Parent(s):
f8551d8
Update pipeline.py
Browse files- pipeline.py +3 -3
pipeline.py
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
from diffusers import DiffusionPipeline
|
| 2 |
-
from invert import Inverter
|
| 3 |
-
from generate import Generator
|
| 4 |
-
from utils import init_model, seed_everything, get_frame_ids
|
| 5 |
|
| 6 |
class VidToMePipeline(DiffusionPipeline):
|
| 7 |
def __init__(self, device="cuda", sd_version="2.1", float_precision="fp16", height=512, width=512):
|
|
|
|
| 1 |
from diffusers import DiffusionPipeline
|
| 2 |
+
from .invert import Inverter
|
| 3 |
+
from .generate import Generator
|
| 4 |
+
from .utils import init_model, seed_everything, get_frame_ids
|
| 5 |
|
| 6 |
class VidToMePipeline(DiffusionPipeline):
|
| 7 |
def __init__(self, device="cuda", sd_version="2.1", float_precision="fp16", height=512, width=512):
|