Commit ·
10394e1
1
Parent(s): 002ca81
Switched to relative import
Browse files- MonsterForgeModel.py +2 -2
MonsterForgeModel.py
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
import torch
|
| 2 |
from transformers import PreTrainedModel
|
| 3 |
-
from MonsterForgeSmallConfig import MonsterForgeSmallConfig
|
| 4 |
-
from LightweightGAN import Generator
|
| 5 |
|
| 6 |
|
| 7 |
class MonsterForgeModel(PreTrainedModel):
|
|
|
|
| 1 |
import torch
|
| 2 |
from transformers import PreTrainedModel
|
| 3 |
+
from .MonsterForgeSmallConfig import MonsterForgeSmallConfig
|
| 4 |
+
from .LightweightGAN import Generator
|
| 5 |
|
| 6 |
|
| 7 |
class MonsterForgeModel(PreTrainedModel):
|