wdcqc commited on
Commit
1032cce
1 Parent(s): 30d20b1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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 get_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 = get_tile_data("wfc/{}_64x64.npz".format(tileset))
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"