Commit 0e72b5bc authored by kurumuz's avatar kurumuz

until it breaks

parent 40f41222
......@@ -339,7 +339,9 @@ class StableDiffusionModel(nn.Module):
def from_url(self, url):
#read config url into bytes
url = f"https://{self.config.s3_endpoint}/{self.config.s3_bucket}/{self.config.s3_folder}/config.yaml"
print(url)
s3_file = self.config.s3_folder + "/config.yaml"
print(s3_file)
headers = web.get_s3_secret_headers(endpoint=self.config.s3_endpoint,
access_key=self.config.s3_access_key,
secret_key=self.config.s3_secret_key,
......@@ -351,6 +353,8 @@ class StableDiffusionModel(nn.Module):
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"
print(s3_file)
print(url)
print(f"Downloading model from {url}")
headers = web.get_s3_secret_headers(endpoint=self.config.s3_endpoint,
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