Spaces:
Running
on
T4
Running
on
T4
Update app.py
Browse files
app.py
CHANGED
@@ -39,7 +39,7 @@ def genie (Model, Prompt, negative_prompt, height, width, scale, steps, seed):
|
|
39 |
progress=gr.Progress(track_tqdm=True)
|
40 |
torch.cuda.empty_cache()
|
41 |
model_repo_id = "stabilityai/stable-diffusion-3.5-large-turbo"
|
42 |
-
pipe = DiffusionPipeline.from_pretrained(model_repo_id)
|
43 |
pipe = pipe.to(device)
|
44 |
pipe.enable_model_cpu_offload()
|
45 |
pipe.vae.enable_slicing()
|
|
|
39 |
progress=gr.Progress(track_tqdm=True)
|
40 |
torch.cuda.empty_cache()
|
41 |
model_repo_id = "stabilityai/stable-diffusion-3.5-large-turbo"
|
42 |
+
pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype = torch.float32)
|
43 |
pipe = pipe.to(device)
|
44 |
pipe.enable_model_cpu_offload()
|
45 |
pipe.vae.enable_slicing()
|