heboya8 commited on
Commit
5786976
·
verified ·
1 Parent(s): 9cf0af3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -5,9 +5,10 @@ from diffusers.utils import export_to_video
5
 
6
  # Initialize the diffusion pipeline
7
  pipe = DiffusionPipeline.from_pretrained(
8
- "heboya8/text2video-test-2",
9
  torch_dtype=torch.float16,
10
- variant="fp16"
 
11
  )
12
 
13
  # Optimize for GPU memory
 
5
 
6
  # Initialize the diffusion pipeline
7
  pipe = DiffusionPipeline.from_pretrained(
8
+ "heboya8/text2video-test-2",
9
  torch_dtype=torch.float16,
10
+ # variant="fp16",
11
+ trust_remote_code=True,
12
  )
13
 
14
  # Optimize for GPU memory