7 lines
210 B
Python
7 lines
210 B
Python
|
import torch
|
||
|
|
||
|
class ConversionContext:
|
||
|
def __init__(self, model_path: str, device: str) -> None:
|
||
|
self.model_path = model_path
|
||
|
self.training_device = device
|
||
|
self.map_location = torch.device(device)
|