Spaces:
Sleeping
Sleeping
Nikhil0987
commited on
Commit
•
eb2c06c
1
Parent(s):
22ee45c
Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,7 @@
|
|
1 |
import torch
|
2 |
from diffusers import DiffusionPipeline, DPMSolverMultistepScheduler
|
3 |
from diffusers.utils import export_to_video
|
|
|
4 |
|
5 |
pipe = DiffusionPipeline.from_pretrained("damo-vilab/text-to-video-ms-1.7b", torch_dtype=torch.float16, variant="fp16")
|
6 |
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
|
|
|
1 |
import torch
|
2 |
from diffusers import DiffusionPipeline, DPMSolverMultistepScheduler
|
3 |
from diffusers.utils import export_to_video
|
4 |
+
import streamlit as st
|
5 |
|
6 |
pipe = DiffusionPipeline.from_pretrained("damo-vilab/text-to-video-ms-1.7b", torch_dtype=torch.float16, variant="fp16")
|
7 |
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
|