dn6 HF Staff commited on
Commit
e0e2430
·
verified ·
1 Parent(s): ac3d493

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. app.py +0 -1
  2. requirements.txt +2 -2
app.py CHANGED
@@ -35,7 +35,6 @@ blocks.sub_blocks.insert("image_to_action", image_to_action_block, 0)
35
  pipe = MatrixGameWanModularPipeline(blocks, "diffusers/matrix-game-2-modular")
36
  pipe.load_components(trust_remote_code=True, device_map="cuda", torch_dtype={"default": torch.bfloat16, "vae": torch.float32})
37
 
38
-
39
  @spaces.GPU(120)
40
  def predict(image, prompt):
41
  output = pipe(image=image, prompt=prompt, num_frames=141)
 
35
  pipe = MatrixGameWanModularPipeline(blocks, "diffusers/matrix-game-2-modular")
36
  pipe.load_components(trust_remote_code=True, device_map="cuda", torch_dtype={"default": torch.bfloat16, "vae": torch.float32})
37
 
 
38
  @spaces.GPU(120)
39
  def predict(image, prompt):
40
  output = pipe(image=image, prompt=prompt, num_frames=141)
requirements.txt CHANGED
@@ -8,8 +8,8 @@ imageio==2.37.0
8
  imageio-ffmpeg==0.6.0
9
  safetensors==0.6.2
10
  sentencepiece==0.2.1
11
- torch==2.7.0
12
  torchao==0.12.0
13
- torchvision==0.22.0
14
  transformers==4.55.4
15
  diffusers @ git+https://github.com/huggingface/diffusers.git
 
8
  imageio-ffmpeg==0.6.0
9
  safetensors==0.6.2
10
  sentencepiece==0.2.1
11
+ torch==2.7.1
12
  torchao==0.12.0
13
+ torchvision==0.22.1
14
  transformers==4.55.4
15
  diffusers @ git+https://github.com/huggingface/diffusers.git