Spaces:

jinysun
/

TeLLAgent

Running

jinysun commited on Apr 8, 2025

Commit

53587f9

verified ·

1 Parent(s): ce124b7

Update tool/comget/generator.py

Files changed (1) hide show

tool/comget/generator.py CHANGED Viewed

@@ -165,7 +165,7 @@ def generation(value):
         model = GPT(mconf)
         args.model_weight =   f'{args.csv_name}.pt'
-        model.load_state_dict(torch.load('tool/comget/' + args.model_weight))
         model.to('cpu')
         print('Model loaded')

         model = GPT(mconf)
         args.model_weight =   f'{args.csv_name}.pt'
+        model.load_state_dict(torch.load('tool/comget/' + args.model_weight, map_location=torch.device('cpu')))
         model.to('cpu')
         print('Model loaded')