jadechoghari
commited on
Commit
•
3ace9e3
1
Parent(s):
43b1627
Update pipeline.py
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
@@ -128,7 +128,7 @@ class VidToMePipeline(DiffusionPipeline):
|
|
128 |
'float_precision': "fp16"
|
129 |
})
|
130 |
|
131 |
-
|
132 |
|
133 |
# # Sample usage
|
134 |
# pipeline = VidToMePipeline(device="cuda", sd_version="2.1", float_precision="fp16")
|
|
|
128 |
'float_precision': "fp16"
|
129 |
})
|
130 |
|
131 |
+
return config
|
132 |
|
133 |
# # Sample usage
|
134 |
# pipeline = VidToMePipeline(device="cuda", sd_version="2.1", float_precision="fp16")
|