Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
c216a7b
1
Parent(s):
f91f26a
joblib fix?
Browse files- auto_forge.py +2 -1
- requirements.txt +1 -1
auto_forge.py
CHANGED
|
@@ -582,7 +582,8 @@ def _initialize_heightmap(
|
|
| 582 |
args.layer_height,
|
| 583 |
bgr_tuple,
|
| 584 |
random_seed=random_seed,
|
| 585 |
-
num_threads=
|
|
|
|
| 586 |
init_method="kmeans",
|
| 587 |
cluster_layers=args.num_init_cluster_layers,
|
| 588 |
material_colors=material_colors_np,
|
|
|
|
| 582 |
args.layer_height,
|
| 583 |
bgr_tuple,
|
| 584 |
random_seed=random_seed,
|
| 585 |
+
num_threads=4,
|
| 586 |
+
num_runs=args.num_init_rounds,
|
| 587 |
init_method="kmeans",
|
| 588 |
cluster_layers=args.num_init_cluster_layers,
|
| 589 |
material_colors=material_colors_np,
|
requirements.txt
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
autoforge==1.9.
|
| 2 |
sentry-sdk[huggingface_hub]
|
| 3 |
sentry-sdk[fastapi]
|
| 4 |
wandb
|
|
|
|
| 1 |
+
autoforge==1.9.1
|
| 2 |
sentry-sdk[huggingface_hub]
|
| 3 |
sentry-sdk[fastapi]
|
| 4 |
wandb
|