Spaces:
Runtime error
Runtime error
Upload folder using huggingface_hub
Browse files- app.py +0 -1
- 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.
|
| 12 |
torchao==0.12.0
|
| 13 |
-
torchvision==0.22.
|
| 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
|