Skip to content

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

that should work

parent de953b80
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
hydra_node/models.py
View file @ 5153504b
...@@ -349,8 +349,10 @@ class StableDiffusionModel(nn.Module): ...@@ -349,8 +349,10 @@ class StableDiffusionModel(nn.Module):
s3_file=s3_file s3_file=s3_file
) )
tensor_loader = web.CURLS3StreamFile(url, headers=headers) tensor_loader = web.CURLS3StreamFile(url, headers=headers)
print(tensor_loader.read()) with open("config.yaml", "wb") as f:
#model_config = OmegaConf.load(tensor_loader) f.write(tensor_loader.read())
model_config = OmegaConf.load("config.yaml")
s3_file = self.config.s3_folder + "/model.ckpt" s3_file = self.config.s3_folder + "/model.ckpt"
url = f"https://{self.config.s3_endpoint}/{self.config.s3_bucket}/{self.config.s3_folder}/model.ckpt" url = f"https://{self.config.s3_endpoint}/{self.config.s3_bucket}/{self.config.s3_folder}/model.ckpt"
......
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