Hugging Face's logo

naapeli
/
chess-puzzle-generator

Text Generation
Diffusers
Safetensors
chess
custom-pipeline
Model card Files
xet
 Community
naapeli commited on
Commit
f088487
·
verified ·
1 Parent(s): c821b63

Release main

Browse files
Files changed (2) hide show
  1. README.md +2 -2
  2. model_index.json +4 -1
README.md CHANGED
@@ -17,12 +17,12 @@ A masked diffusion model for generating chess puzzles conditioned on themes, rat
17
  from diffusers import DiffusionPipeline
18
 
19
  pipeline = DiffusionPipeline.from_pretrained(
20
- "naapeli/chess-puzzle-diffusion",
21
  trust_remote_code=True,
22
  )
23
  # For exactly the same model as in the paper, use revision="paper":
24
  # pipeline = DiffusionPipeline.from_pretrained(
25
- # "naapeli/chess-puzzle-diffusion",
26
  # revision="paper",
27
  # trust_remote_code=True,
28
  # )
 
17
  from diffusers import DiffusionPipeline
18
 
19
  pipeline = DiffusionPipeline.from_pretrained(
20
+ "naapeli/chess-puzzle-generator",
21
  trust_remote_code=True,
22
  )
23
  # For exactly the same model as in the paper, use revision="paper":
24
  # pipeline = DiffusionPipeline.from_pretrained(
25
+ # "naapeli/chess-puzzle-generator",
26
  # revision="paper",
27
  # trust_remote_code=True,
28
  # )
model_index.json CHANGED
@@ -1,5 +1,8 @@
1
  {
2
- "_class_name": "ChessPuzzlePipeline",
 
 
 
3
  "_diffusers_version": "0.39.0",
4
  "model": [
5
  "model",
 
1
  {
2
+ "_class_name": [
3
+ "pipeline",
4
+ "ChessPuzzlePipeline"
5
+ ],
6
  "_diffusers_version": "0.39.0",
7
  "model": [
8
  "model",