Hugging Face's logo

SupraLabs
/
SupraMNST-IMG-200k

Text-to-Image
Safetensors
digit_diffusion
harley-ml
image
digit-to-image
mnist
small
supralabs
custom_code
Model card Files
xet
 Community
Harley-ml commited on
Commit
39c903a
·
verified ·
1 Parent(s): 4e76fed

Update modeling.py

Browse files
Files changed (1) hide show
  1. modeling.py +2 -2
modeling.py CHANGED
@@ -1,5 +1,5 @@
1
  #!/usr/bin/env python3
2
- # Model for MNiST-IMG-390k
3
 
4
  from __future__ import annotations
5
 
@@ -11,7 +11,7 @@ from diffusers import UNet2DConditionModel
11
  from transformers import PreTrainedModel
12
  from transformers.utils import ModelOutput
13
 
14
- from configuration import DigitDiffusionConfig
15
 
16
 
17
  @dataclass
 
1
  #!/usr/bin/env python3
2
+ # Model for SupraMNiST-IMG-200k
3
 
4
  from __future__ import annotations
5
 
 
11
  from transformers import PreTrainedModel
12
  from transformers.utils import ModelOutput
13
 
14
+ from .configuration import DigitDiffusionConfig
15
 
16
 
17
  @dataclass