image_through_model #49
@@ -23,7 +23,6 @@ def get_model(
|
||||
client=client,
|
||||
object_name=object_name,
|
||||
)
|
||||
logging.debug('buffer size: %s', buffer.getbuffer().nbytes)
|
||||
# convert data to model checkpoint
|
||||
buffer.seek(0)
|
||||
model_content = torch.load(buffer)
|
||||
|
||||
Reference in New Issue
Block a user