Update app.py
Browse files
app.py
CHANGED
|
@@ -10,7 +10,7 @@ from generator import Generator, LATENT_DIM
|
|
| 10 |
import math
|
| 11 |
|
| 12 |
generator = Generator(7)
|
| 13 |
-
variables = generator.init(jax.random.PRNGKey(0), jnp.zeros([1, LATENT_DIM])
|
| 14 |
|
| 15 |
fs = HfFileSystem()
|
| 16 |
with fs.open("PrakhAI/AIPlane3/g_checkpoint_400000.msgpack", "rb") as f:
|
|
|
|
| 10 |
import math
|
| 11 |
|
| 12 |
generator = Generator(7)
|
| 13 |
+
variables = generator.init(jax.random.PRNGKey(0), jnp.zeros([1, LATENT_DIM]))
|
| 14 |
|
| 15 |
fs = HfFileSystem()
|
| 16 |
with fs.open("PrakhAI/AIPlane3/g_checkpoint_400000.msgpack", "rb") as f:
|