lopho commited on
Commit
46fb568
1 Parent(s): b83ebfb

forgot to change non-local model path

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ _preheat: bool = False
21
  _seen_compilations = set()
22
 
23
  _model = InferenceUNetPseudo3D(
24
- model_path = '/mnt/work1/make_a_vid/makeavid-space/model/model',
25
  dtype = jnp.float16,
26
  hf_auth_token = os.environ.get('HUGGING_FACE_HUB_TOKEN', None)
27
  )
 
21
  _seen_compilations = set()
22
 
23
  _model = InferenceUNetPseudo3D(
24
+ model_path = 'TempoFunk/makeavid-sd-jax',
25
  dtype = jnp.float16,
26
  hf_auth_token = os.environ.get('HUGGING_FACE_HUB_TOKEN', None)
27
  )