Update README.md
Browse files
README.md
CHANGED
|
@@ -62,7 +62,7 @@ import torch
|
|
| 62 |
from transformers import AutoTokenizer
|
| 63 |
|
| 64 |
# Load tokenizer
|
| 65 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
| 66 |
|
| 67 |
# Load model (see inference.py for full diffusion generation logic)
|
| 68 |
# The model uses custom DiffusionQwen3Model class
|
|
|
|
| 62 |
from transformers import AutoTokenizer
|
| 63 |
|
| 64 |
# Load tokenizer
|
| 65 |
+
tokenizer = AutoTokenizer.from_pretrained("shouryamaanjain/adhd-diffusion", trust_remote_code=True)
|
| 66 |
|
| 67 |
# Load model (see inference.py for full diffusion generation logic)
|
| 68 |
# The model uses custom DiffusionQwen3Model class
|