Commit
•
a6dbfcc
1
Parent(s):
704dc77
Update pipeline.py
Browse files- pipeline.py +2 -1
pipeline.py
CHANGED
|
@@ -126,7 +126,8 @@ class VidToMePipeline(DiffusionPipeline):
|
|
| 126 |
},
|
| 127 |
'seed': seed,
|
| 128 |
'device': "cuda",
|
| 129 |
-
'float_precision': "fp16"
|
|
|
|
| 130 |
})
|
| 131 |
|
| 132 |
return config
|
|
|
|
| 126 |
},
|
| 127 |
'seed': seed,
|
| 128 |
'device': "cuda",
|
| 129 |
+
'float_precision': "fp16",
|
| 130 |
+
'enable_xformers_memory_efficient_attention': True
|
| 131 |
})
|
| 132 |
|
| 133 |
return config
|