hvoss-techfak commited on
Commit
c216a7b
·
1 Parent(s): f91f26a

joblib fix?

Browse files
Files changed (2) hide show
  1. auto_forge.py +2 -1
  2. 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=args.num_init_rounds,
 
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.0
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