lopho commited on
Commit
8905bfd
1 Parent(s): 1ebccde
Files changed (2) hide show
  1. README.md +2 -0
  2. makeavid_sd/inference.py +2 -2
README.md CHANGED
@@ -8,6 +8,8 @@ sdk_version: 3.28.0
8
  app_file: app.py
9
  pinned: true
10
  license: agpl-3.0
 
 
11
  ---
12
 
13
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
8
  app_file: app.py
9
  pinned: true
10
  license: agpl-3.0
11
+ tags:
12
+ - jax-diffusers-event
13
  ---
14
 
15
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
makeavid_sd/inference.py CHANGED
@@ -121,7 +121,7 @@ class InferenceUNetPseudo3D:
121
  self.model_path,
122
  subfolder = 'scheduler',
123
  dtype = jnp.float32,
124
- use_auth_token = self.hf_api_key
125
  )
126
  self.scheduler: scheduler_cls = scheduler
127
  self.params['scheduler'] = scheduler_state
@@ -134,7 +134,7 @@ class InferenceUNetPseudo3D:
134
  self.model_path,
135
  subfolder = 'scheduler',
136
  dtype = jnp.float32,
137
- use_auth_token = self.hf_api_key
138
  )
139
  self.scheduler: scheduler_cls = scheduler
140
  self.params['scheduler'] = scheduler_state
 
121
  self.model_path,
122
  subfolder = 'scheduler',
123
  dtype = jnp.float32,
124
+ use_auth_token = self.hf_auth_token
125
  )
126
  self.scheduler: scheduler_cls = scheduler
127
  self.params['scheduler'] = scheduler_state
 
134
  self.model_path,
135
  subfolder = 'scheduler',
136
  dtype = jnp.float32,
137
+ use_auth_token = self.hf_auth_token
138
  )
139
  self.scheduler: scheduler_cls = scheduler
140
  self.params['scheduler'] = scheduler_state