image_through_model #49

Merged
brian merged 75 commits from image_through_model into main 2024-10-20 00:10:27 +02:00
Showing only changes of commit e698d058ff - Show all commits
-1
View File
@@ -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)