stellaathena commited on
Commit
28520b2
1 Parent(s): a9978c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
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",