Spaces:
Starting
on
A10G
Starting
on
A10G
Merge branch 'main' of https://huggingface.co/spaces/baulab/ConceptSliders into main
Browse files
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()
|