PrakhAI commited on
Commit
d33fcf2
·
1 Parent(s): 41451f5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ generator = Generator(7)
13
  variables = generator.init(jax.random.PRNGKey(0), jnp.zeros([1, LATENT_DIM]), training=False)
14
 
15
  fs = HfFileSystem()
16
- with fs.open("PrakhAI/AIPlane3/g_checkpoint_200000.msgpack", "rb") as f:
17
  g_state = from_state_dict(variables, msgpack_restore(f.read()))
18
 
19
  def sample_latent(batch, key):
 
13
  variables = generator.init(jax.random.PRNGKey(0), jnp.zeros([1, LATENT_DIM]), training=False)
14
 
15
  fs = HfFileSystem()
16
+ with fs.open("PrakhAI/AIPlane3/g_checkpoint_400000.msgpack", "rb") as f:
17
  g_state = from_state_dict(variables, msgpack_restore(f.read()))
18
 
19
  def sample_latent(batch, key):