Spaces:
Runtime error
Runtime error
Torch 16
Browse files- text_to_video.py +1 -1
text_to_video.py
CHANGED
@@ -37,7 +37,7 @@ class TextToVideoTool(Tool):
|
|
37 |
self.device = get_default_device()
|
38 |
|
39 |
self.pipeline = DiffusionPipeline.from_pretrained(
|
40 |
-
self.default_checkpoint,
|
41 |
)
|
42 |
self.pipeline.to(self.device)
|
43 |
|
|
|
37 |
self.device = get_default_device()
|
38 |
|
39 |
self.pipeline = DiffusionPipeline.from_pretrained(
|
40 |
+
self.default_checkpoint, variant="fp16"
|
41 |
)
|
42 |
self.pipeline.to(self.device)
|
43 |
|