Spaces:
Running
Running
Commit
·
0bb12d1
1
Parent(s):
3039940
options/Video_model/Model.py
CHANGED
|
@@ -3,6 +3,7 @@ from diffusers import StableVideoDiffusionPipeline
|
|
| 3 |
from diffusers.utils import load_image, export_to_video
|
| 4 |
from PIL import Image
|
| 5 |
from huggingface_hub import login
|
|
|
|
| 6 |
login(token=os.getenv("TOKEN"))
|
| 7 |
# Check if CUDA (GPU) is available, otherwise use CPU
|
| 8 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
|
| 3 |
from diffusers.utils import load_image, export_to_video
|
| 4 |
from PIL import Image
|
| 5 |
from huggingface_hub import login
|
| 6 |
+
import numpy as np
|
| 7 |
login(token=os.getenv("TOKEN"))
|
| 8 |
# Check if CUDA (GPU) is available, otherwise use CPU
|
| 9 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
options/Video_model/__pycache__/Model.cpython-310.pyc
CHANGED
|
Binary files a/options/Video_model/__pycache__/Model.cpython-310.pyc and b/options/Video_model/__pycache__/Model.cpython-310.pyc differ
|
|
|