Spaces:

AlyxTeam
/

Realistic_Vision_V6.0_B1_noVAE

Running on Zero

AlyxTeam commited on Sep 19

Commit

3f78aaf

•

1 Parent(s): b8543c8

feat: 使用GPU

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import gradio as gr
 import numpy as np
 import random
-#import spaces #[uncomment to use ZeroGPU]
 from diffusers import DiffusionPipeline
 import torch
@@ -19,7 +19,7 @@ pipe = pipe.to(device)
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
-#@spaces.GPU #[uncomment to use ZeroGPU]
 def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps, progress=gr.Progress(track_tqdm=True)):
     if randomize_seed:

 import gradio as gr
 import numpy as np
 import random
+import spaces #[uncomment to use ZeroGPU]
 from diffusers import DiffusionPipeline
 import torch
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
+@spaces.GPU #[uncomment to use ZeroGPU]
 def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps, progress=gr.Progress(track_tqdm=True)):
     if randomize_seed: