Upload base_v2.py

base_v2.py

@@ -153,6 +153,7 @@ class ConfiguredModelArchitecture(BaseModel, frozen=True, arbitrary_types_allowe
             weight_prefix=self.info.modules[module_name].weight_prefix,
         )
 
+# Runpod Fix
 # Manually rebuild Pydantic models to resolve forward references
 # This fixes the "not fully defined" error with Pydantic v2
 ConfiguredModuleArchitecture.model_rebuild()