Update README.md
Browse files
README.md
CHANGED
@@ -61,13 +61,13 @@ Then create a Jupyter notebook under the repository folder:
|
|
61 |
# Load pipeline
|
62 |
from wfd.wf_diffusers import WaveFunctionDiffusionPipeline
|
63 |
from wfd.wf_diffusers import AutoencoderTile
|
64 |
-
from wfd.scmap import
|
65 |
|
66 |
# Tilesets: ashworld, badlands, desert, ice, jungle, platform, twilight, install
|
67 |
tileset = "ice"
|
68 |
|
69 |
# The data files are located in wfd/scmap/tile_data/wfc
|
70 |
-
wfc_data_path =
|
71 |
|
72 |
# Use CUDA (otherwise it will take 15 minutes)
|
73 |
device = "cuda"
|
|
|
61 |
# Load pipeline
|
62 |
from wfd.wf_diffusers import WaveFunctionDiffusionPipeline
|
63 |
from wfd.wf_diffusers import AutoencoderTile
|
64 |
+
from wfd.scmap import find_tile_data, get_tileset_keyword
|
65 |
|
66 |
# Tilesets: ashworld, badlands, desert, ice, jungle, platform, twilight, install
|
67 |
tileset = "ice"
|
68 |
|
69 |
# The data files are located in wfd/scmap/tile_data/wfc
|
70 |
+
wfc_data_path = find_tile_data("wfc/{}_64x64.npz".format(tileset))
|
71 |
|
72 |
# Use CUDA (otherwise it will take 15 minutes)
|
73 |
device = "cuda"
|