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): ...@@ -346,7 +346,7 @@ class StableDiffusionModel(nn.Module):
bucket=self.config.s3_bucket, bucket=self.config.s3_bucket,
s3_file=s3_file s3_file=s3_file
) )
tensor_loader = web.CURLStreamFile(url, headers=headers) tensor_loader = web.CURLS3StreamFile(url, headers=headers)
model_config = OmegaConf.load(tensor_loader) model_config = OmegaConf.load(tensor_loader)
s3_file = self.config.s3_folder + "/model.ckpt" s3_file = self.config.s3_folder + "/model.ckpt"
...@@ -358,7 +358,7 @@ class StableDiffusionModel(nn.Module): ...@@ -358,7 +358,7 @@ class StableDiffusionModel(nn.Module):
bucket=self.config.s3_bucket, bucket=self.config.s3_bucket,
s3_file=s3_file 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) model = self.load_model_from_config(model_config, tensor_loader)
return model, model_config 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