Hugging Face's logo

Spaces:
Indulge-Bai
/
Weak-to-Strong-Diffusion
Running on Zero

App Files Community
1
Indulge-Bai commited on
Commit
07075bd
·
verified ·
1 Parent(s): 8002483

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import gradio as gr
2
  import torch
3
  from diffusers import UNet2DConditionModel, DDIMInverseScheduler, DDIMScheduler
@@ -5,7 +6,6 @@ from utils.pipeline_stable_diffusion_xl import StableDiffusionXLPipeline
5
  import torch
6
  from PIL import Image
7
  import argparse
8
- import spaces
9
 
10
  weak_model = StableDiffusionXLPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0")
11
  strong_model = StableDiffusionXLPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0")
 
1
+ import spaces
2
  import gradio as gr
3
  import torch
4
  from diffusers import UNet2DConditionModel, DDIMInverseScheduler, DDIMScheduler
 
6
  import torch
7
  from PIL import Image
8
  import argparse
 
9
 
10
  weak_model = StableDiffusionXLPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0")
11
  strong_model = StableDiffusionXLPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0")