Skip to content

H
Hydra Node Http
Commit b8f4ae93 authored by kurumuz's avatar kurumuz
Browse files
Options

im sorry

parent dc48673d
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
hydra_node/models.py
View file @ b8f4ae93
......@@ -346,7 +346,7 @@ class StableDiffusionModel(nn.Module):
bucket=self.config.s3_bucket,
s3_file=s3_file
)
tensor_loader = web.CURLStreamFile(url, headers=headers)
tensor_loader = web.CURLS3StreamFile(url, headers=headers)
model_config = OmegaConf.load(tensor_loader)
s3_file = self.config.s3_folder + "/model.ckpt"
......@@ -358,7 +358,7 @@ class StableDiffusionModel(nn.Module):
bucket=self.config.s3_bucket,
s3_file=s3_file
)
tensor_loader = web.CURLStreamFile(url, headers=headers)
tensor_loader = web.CURLS3StreamFile(url, headers=headers)
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