mirror of
https://github.com/deepseek-ai/DreamCraft3D
synced 2025-06-26 18:25:49 +00:00
chores: rebase commits
This commit is contained in:
36
threestudio/__init__.py
Normal file
36
threestudio/__init__.py
Normal file
@@ -0,0 +1,36 @@
|
||||
__modules__ = {}
|
||||
|
||||
|
||||
def register(name):
|
||||
def decorator(cls):
|
||||
__modules__[name] = cls
|
||||
return cls
|
||||
|
||||
return decorator
|
||||
|
||||
|
||||
def find(name):
|
||||
return __modules__[name]
|
||||
|
||||
|
||||
### grammar sugar for logging utilities ###
|
||||
import logging
|
||||
|
||||
logger = logging.getLogger("pytorch_lightning")
|
||||
|
||||
from pytorch_lightning.utilities.rank_zero import (
|
||||
rank_zero_debug,
|
||||
rank_zero_info,
|
||||
rank_zero_only,
|
||||
)
|
||||
|
||||
debug = rank_zero_debug
|
||||
info = rank_zero_info
|
||||
|
||||
|
||||
@rank_zero_only
|
||||
def warn(*args, **kwargs):
|
||||
logger.warn(*args, **kwargs)
|
||||
|
||||
|
||||
from . import data, models, systems
|
||||
Reference in New Issue
Block a user