Spaces:
Runtime error
Runtime error
Commit
·
28520b2
1
Parent(s):
a9978c2
Update app.py
Browse files
app.py
CHANGED
|
@@ -3,6 +3,9 @@ import re
|
|
| 3 |
from magma import Magma
|
| 4 |
from magma.image_input import ImageInput
|
| 5 |
|
|
|
|
|
|
|
|
|
|
| 6 |
model = Magma.from_checkpoint(
|
| 7 |
config_path = "configs/MAGMA_v1.yml",
|
| 8 |
checkpoint_path = "./mp_rank_00_model_states.pt",
|
|
|
|
| 3 |
from magma import Magma
|
| 4 |
from magma.image_input import ImageInput
|
| 5 |
|
| 6 |
+
import os
|
| 7 |
+
os.system("pip install deepspeed")
|
| 8 |
+
|
| 9 |
model = Magma.from_checkpoint(
|
| 10 |
config_path = "configs/MAGMA_v1.yml",
|
| 11 |
checkpoint_path = "./mp_rank_00_model_states.pt",
|