Spaces:
Runtime error
Runtime error
stellaathena
commited on
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",
|