Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -15,7 +15,7 @@ def genie (Model, Prompt, negative_prompt, height, width, scale, steps, seed):
|
|
15 |
if Model == "SD3":
|
16 |
#torch.cuda.max_memory_allocated(device=device)
|
17 |
torch.cuda.empty_cache()
|
18 |
-
SD3 = QwenImagePipeline.from_pretrained("Qwen/Qwen-Image", torch_dtype=torch.
|
19 |
torch.cuda.empty_cache()
|
20 |
image=SD3(
|
21 |
prompt=Prompt,
|
|
|
15 |
if Model == "SD3":
|
16 |
#torch.cuda.max_memory_allocated(device=device)
|
17 |
torch.cuda.empty_cache()
|
18 |
+
SD3 = QwenImagePipeline.from_pretrained("Qwen/Qwen-Image", torch_dtype=torch.float16).to(device)
|
19 |
torch.cuda.empty_cache()
|
20 |
image=SD3(
|
21 |
prompt=Prompt,
|