Spaces:
Running
on
T4
Running
on
T4
Update app.py
Browse files
app.py
CHANGED
@@ -301,7 +301,7 @@ def genie (Model, Prompt, negative_prompt, height, width, scale, steps, seed, re
|
|
301 |
image = sdxl(Prompt, negative_prompt=negative_prompt, height=height, width=width, num_inference_steps=steps, guidance_scale=scale).images[0]
|
302 |
torch.cuda.empty_cache()
|
303 |
|
304 |
-
if
|
305 |
|
306 |
pipe = DiffusionPipeline.from_pretrained("circulus/canvers-fusionXL-v1", torch_dtype=torch.float16, safety_checker=None) if torch.cuda.is_available() else DiffusionPipeline.from_pretrained("circulus/canvers-real-v3.8.1")
|
307 |
pipe.enable_xformers_memory_efficient_attention()
|
|
|
301 |
image = sdxl(Prompt, negative_prompt=negative_prompt, height=height, width=width, num_inference_steps=steps, guidance_scale=scale).images[0]
|
302 |
torch.cuda.empty_cache()
|
303 |
|
304 |
+
if Model == 'Test':
|
305 |
|
306 |
pipe = DiffusionPipeline.from_pretrained("circulus/canvers-fusionXL-v1", torch_dtype=torch.float16, safety_checker=None) if torch.cuda.is_available() else DiffusionPipeline.from_pretrained("circulus/canvers-real-v3.8.1")
|
307 |
pipe.enable_xformers_memory_efficient_attention()
|