Spaces:
Running
Running
zR
commited on
Commit
•
4873975
1
Parent(s):
4bbaf2d
using with float16
Browse files
app.py
CHANGED
|
@@ -10,7 +10,7 @@ from datetime import datetime, timedelta
|
|
| 10 |
from openai import OpenAI
|
| 11 |
import moviepy.editor as mp
|
| 12 |
|
| 13 |
-
dtype = torch.
|
| 14 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 15 |
pipe = CogVideoXPipeline.from_pretrained("THUDM/CogVideoX-2b", torch_dtype=dtype).to(device)
|
| 16 |
|
|
|
|
| 10 |
from openai import OpenAI
|
| 11 |
import moviepy.editor as mp
|
| 12 |
|
| 13 |
+
dtype = torch.float16
|
| 14 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 15 |
pipe = CogVideoXPipeline.from_pretrained("THUDM/CogVideoX-2b", torch_dtype=dtype).to(device)
|
| 16 |
|