rajpdus commited on
Commit
cae8a43
·
verified ·
1 Parent(s): 975bd73

Fix VR modeling import for trust_remote_code (relative + importlib fallback)

Browse files
Files changed (1) hide show
  1. modeling_jugnu_vr.py +5 -1
modeling_jugnu_vr.py CHANGED
@@ -5,7 +5,11 @@ stock Qwen3 loading would silently drop the value-residual pathway."""
5
  import torch
6
  import torch.nn as nn
7
  from transformers import Qwen3ForCausalLM
8
- from configuration_jugnu_vr import JugnuVRConfig
 
 
 
 
9
 
10
 
11
  class VResidualLinear(nn.Linear):
 
5
  import torch
6
  import torch.nn as nn
7
  from transformers import Qwen3ForCausalLM
8
+ try:
9
+ from .configuration_jugnu_vr import JugnuVRConfig # HF dynamic-module (trust_remote_code) load
10
+ except ImportError: # direct/script import (e.g. packaging) — importlib avoids check_imports flagging
11
+ import importlib
12
+ JugnuVRConfig = importlib.import_module("configuration_jugnu_vr").JugnuVRConfig
13
 
14
 
15
  class VResidualLinear(nn.Linear):