Skip to content

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

maybe...

parent 723abd56
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
hydra_node/models.py
View file @ 40f41222
...@@ -338,7 +338,7 @@ class StableDiffusionModel(nn.Module): ...@@ -338,7 +338,7 @@ class StableDiffusionModel(nn.Module):
def from_url(self, url): def from_url(self, url):
#read config url into bytes #read config url into bytes
url = f"https://{self.config.s3_bucket}/{self.config.s3_folder}/config.yaml" url = f"https://{self.config.s3_endpoint}/{self.config.s3_bucket}/{self.config.s3_folder}/config.yaml"
s3_file = self.config.s3_folder + "/config.yaml" s3_file = self.config.s3_folder + "/config.yaml"
headers = web.get_s3_secret_headers(endpoint=self.config.s3_endpoint, headers = web.get_s3_secret_headers(endpoint=self.config.s3_endpoint,
access_key=self.config.s3_access_key, access_key=self.config.s3_access_key,
...@@ -350,7 +350,7 @@ class StableDiffusionModel(nn.Module): ...@@ -350,7 +350,7 @@ class StableDiffusionModel(nn.Module):
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"
url = f"https://{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"
print(f"Downloading model from {url}") print(f"Downloading model from {url}")
headers = web.get_s3_secret_headers(endpoint=self.config.s3_endpoint, headers = web.get_s3_secret_headers(endpoint=self.config.s3_endpoint,
access_key=self.config.s3_access_key, access_key=self.config.s3_access_key,
......
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