Spaces:

vllab
/

controlnet-hands

Runtime error

vumichien commited on May 8, 2023

Commit

bf39427

1 Parent(s): 81e463a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -154,7 +154,7 @@ enc_pipeline_params = jax_utils.replicate(enc_pipeline_params)
 rng = jax.random.PRNGKey(0)
 num_samples = jax.device_count()
 prng_seed = jax.random.split(rng, jax.device_count())
 def infer(prompt, negative_prompt, image, model_type="Standard"):
     prompts = num_samples * [prompt]

 rng = jax.random.PRNGKey(0)
 num_samples = jax.device_count()
 prng_seed = jax.random.split(rng, jax.device_count())
+memory = psutil.virtual_memory()
 def infer(prompt, negative_prompt, image, model_type="Standard"):
     prompts = num_samples * [prompt]