Skip to content

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

try this

parent 116cf483
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
hydra_node/models.py
View file @ de953b80
...@@ -349,7 +349,8 @@ class StableDiffusionModel(nn.Module): ...@@ -349,7 +349,8 @@ 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)
model_config = OmegaConf.load(tensor_loader) print(tensor_loader.read())
#model_config = OmegaConf.load(tensor_loader)
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