Commit
•
526c5d8
1
Parent(s):
629f20f
Update pipeline.py
Browse files- pipeline.py +1 -0
pipeline.py
CHANGED
|
@@ -2,6 +2,7 @@ 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):
|
|
|
|
| 2 |
from .invert import Inverter
|
| 3 |
from .generate import Generator
|
| 4 |
from .utils import init_model, seed_everything, get_frame_ids
|
| 5 |
+
import torch
|
| 6 |
|
| 7 |
class VidToMePipeline(DiffusionPipeline):
|
| 8 |
# def __init__(self, device="cuda", sd_version="2.1", float_precision="fp16", height=512, width=512):
|