fix set model attr

[ccssu]

来源： #675

• use_graph=False
• use_graph=True

test.py

import torch 
from torch.nn import Module
import oneflow as flow
from onediff.infer_compiler import oneflow_compile
from onediff.infer_compiler.transform import register

class Demo(Module):
    def __init__(self):
        super(Demo, self).__init__()
        self.value = 0

    def forward(self, x):
        return x * self.value
    
    def set_value(self, value):
        self.value = value

class Demo2(flow.nn.Module):
    def __init__(self):
        super(Demo2, self).__init__()
        self.value = 0

    def forward(self, x):
        return x * self.value
    
    def set_value(self, value):
        self.value = value

register(torch2oflow_class_map={Demo: Demo2})

model = Demo()
of_model = oneflow_compile(model, use_graph=False)
input_data = torch.tensor([1.0])

model.set_value(1.0)
of_output = of_model(input_data)
pt_output = model(input_data)

print("OneFlow Output:", of_output)
print("PyTorch Output:", pt_output)
# OneFlow Output: tensor([1.])
# PyTorch Output: tensor([1.])