Update app.py
Browse files
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/
|
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):
|