RohitGandikota commited on
Commit
ea4edf5
·
2 Parent(s): a3a1429 f97034c

Merge branch 'main' of https://huggingface.co/spaces/baulab/ConceptSliders into main

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -4,7 +4,8 @@ import os
4
  from utils import call
5
  from diffusers.pipelines import StableDiffusionXLPipeline
6
  StableDiffusionXLPipeline.__call__ = call
7
-
 
8
  model_map = {'Age' : 'models/age.pt',
9
  'Chubby': 'models/chubby.pt',
10
  'Muscular': 'models/muscular.pt',
@@ -35,7 +36,7 @@ class Demo:
35
  self.generating = False
36
  self.device = 'cuda'
37
  self.weight_dtype = torch.float16
38
- self.pipe = StableDiffusionXLPipeline.from_pretrained('stabilityai/stable-diffusion-xl-base-1.0', torch_dtype=weight_dtype)
39
 
40
  with gr.Blocks() as demo:
41
  self.layout()
 
4
  from utils import call
5
  from diffusers.pipelines import StableDiffusionXLPipeline
6
  StableDiffusionXLPipeline.__call__ = call
7
+ import os
8
+ os.environ['CURL_CA_BUNDLE'] = ''
9
  model_map = {'Age' : 'models/age.pt',
10
  'Chubby': 'models/chubby.pt',
11
  'Muscular': 'models/muscular.pt',
 
36
  self.generating = False
37
  self.device = 'cuda'
38
  self.weight_dtype = torch.float16
39
+ self.pipe = StableDiffusionXLPipeline.from_pretrained('stabilityai/stable-diffusion-xl-base-1.0', torch_dtype=self.weight_dtype)
40
 
41
  with gr.Blocks() as demo:
42
  self.layout()