Upload ConditionalUNet
Browse files- model.safetensors +1 -1
- modeling_conditional_unet.py +2 -2
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 293858844
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:921df775e45cf6b55d1b2cedd5397e8c9556b0e22f25d91d986931b7ad0392e5
|
| 3 |
size 293858844
|
modeling_conditional_unet.py
CHANGED
|
@@ -27,11 +27,11 @@ class UpSampleBlock(nn.Module):
|
|
| 27 |
return x
|
| 28 |
|
| 29 |
class ConditionalUNet(PreTrainedModel):
|
| 30 |
-
config_class =
|
| 31 |
|
| 32 |
def __init__(self, config):
|
| 33 |
super().__init__(config)
|
| 34 |
-
self.config_class = 'configuration_conditional_unet.ConditionalUNetConfig'
|
| 35 |
self.config = config
|
| 36 |
|
| 37 |
self.encoder_rep = config.encoder_rep
|
|
|
|
| 27 |
return x
|
| 28 |
|
| 29 |
class ConditionalUNet(PreTrainedModel):
|
| 30 |
+
config_class = ConditionalUNetConfig
|
| 31 |
|
| 32 |
def __init__(self, config):
|
| 33 |
super().__init__(config)
|
| 34 |
+
# self.config_class = 'configuration_conditional_unet.ConditionalUNetConfig'
|
| 35 |
self.config = config
|
| 36 |
|
| 37 |
self.encoder_rep = config.encoder_rep
|