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):
|