Commit 74c57c46 authored by kurumuz's avatar kurumuz

should work

parent 5153504b
......@@ -366,6 +366,7 @@ class StableDiffusionModel(nn.Module):
s3_file=s3_file
)
tensor_loader = web.CURLS3StreamFile(url, headers=headers)
tensor_loader = io.BytesIO(tensor_loader.read())
model = self.load_model_from_config(model_config, tensor_loader)
return model, model_config
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment