Spaces:
Running
on
Zero
Running
on
Zero
Update
Browse files
README.md
CHANGED
@@ -4,7 +4,7 @@ emoji: 🏢
|
|
4 |
colorFrom: blue
|
5 |
colorTo: pink
|
6 |
sdk: gradio
|
7 |
-
sdk_version: 3.
|
8 |
app_file: app.py
|
9 |
pinned: false
|
10 |
suggested_hardware: t4-small
|
|
|
4 |
colorFrom: blue
|
5 |
colorTo: pink
|
6 |
sdk: gradio
|
7 |
+
sdk_version: 3.36.1
|
8 |
app_file: app.py
|
9 |
pinned: false
|
10 |
suggested_hardware: t4-small
|
model.py
CHANGED
@@ -19,8 +19,6 @@ sys.path.insert(0, 'multires_textual_inversion')
|
|
19 |
|
20 |
from pipeline import MultiResPipeline, load_learned_concepts
|
21 |
|
22 |
-
HF_TOKEN = os.getenv('HF_TOKEN')
|
23 |
-
|
24 |
|
25 |
class Model:
|
26 |
def __init__(self):
|
@@ -28,13 +26,11 @@ class Model:
|
|
28 |
'cuda:0' if torch.cuda.is_available() else 'cpu')
|
29 |
model_id = 'runwayml/stable-diffusion-v1-5'
|
30 |
if self.device.type == 'cpu':
|
31 |
-
pipe = MultiResPipeline.from_pretrained(model_id
|
32 |
-
use_auth_token=HF_TOKEN)
|
33 |
else:
|
34 |
pipe = MultiResPipeline.from_pretrained(model_id,
|
35 |
torch_dtype=torch.float16,
|
36 |
-
revision='fp16'
|
37 |
-
use_auth_token=HF_TOKEN)
|
38 |
self.pipe = pipe.to(self.device)
|
39 |
self.pipe.scheduler = DPMSolverMultistepScheduler(
|
40 |
beta_start=0.00085,
|
|
|
19 |
|
20 |
from pipeline import MultiResPipeline, load_learned_concepts
|
21 |
|
|
|
|
|
22 |
|
23 |
class Model:
|
24 |
def __init__(self):
|
|
|
26 |
'cuda:0' if torch.cuda.is_available() else 'cpu')
|
27 |
model_id = 'runwayml/stable-diffusion-v1-5'
|
28 |
if self.device.type == 'cpu':
|
29 |
+
pipe = MultiResPipeline.from_pretrained(model_id)
|
|
|
30 |
else:
|
31 |
pipe = MultiResPipeline.from_pretrained(model_id,
|
32 |
torch_dtype=torch.float16,
|
33 |
+
revision='fp16')
|
|
|
34 |
self.pipe = pipe.to(self.device)
|
35 |
self.pipe.scheduler = DPMSolverMultistepScheduler(
|
36 |
beta_start=0.00085,
|