Spaces:
Running
on
T4
Running
on
T4
Update app.py
Browse files
app.py
CHANGED
@@ -170,7 +170,7 @@ def genie (Model, Prompt, negative_prompt, height, width, scale, steps, seed, re
|
|
170 |
return image
|
171 |
|
172 |
if Model == "SemiReal":
|
173 |
-
semi = DiffusionPipeline.from_pretrained("circulus/canvers-semi-v3.
|
174 |
semi.enable_xformers_memory_efficient_attention()
|
175 |
semi = semi.to(device)
|
176 |
torch.cuda.empty_cache()
|
|
|
170 |
return image
|
171 |
|
172 |
if Model == "SemiReal":
|
173 |
+
semi = DiffusionPipeline.from_pretrained("circulus/canvers-semi-v3.8.1", torch_dtype=torch.float16, safety_checker=None) if torch.cuda.is_available() else DiffusionPipeline.from_pretrained("circulus/canvers-semi-v3.8.1")
|
174 |
semi.enable_xformers_memory_efficient_attention()
|
175 |
semi = semi.to(device)
|
176 |
torch.cuda.empty_cache()
|