Falln87 commited on
Commit
9422171
1 Parent(s): 265cdf3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import streamlit as st
2
  from PIL import Image
3
  from diffusers import StableDiffusionPipeline, ControlNetModel, DDIMScheduler, LMSDiscreteScheduler, UNet2DConditionModel, DiffusionPipeline
4
- from diffusers.optimization import DDPMScheduler, DDPMSchedulerV2, PNDMScheduler
5
  from transformers import AutoModelForCausalLM, AutoTokenizer, AutoModelForSeq2SeqLM, LlamaTokenizerFast, LlamaForCausalLM
6
  from accelerate import Accelerator
7
  import torch
 
1
  import streamlit as st
2
  from PIL import Image
3
  from diffusers import StableDiffusionPipeline, ControlNetModel, DDIMScheduler, LMSDiscreteScheduler, UNet2DConditionModel, DiffusionPipeline
4
+ from diffusers import DDPMScheduler, DDPMSchedulerV2, PNDMScheduler
5
  from transformers import AutoModelForCausalLM, AutoTokenizer, AutoModelForSeq2SeqLM, LlamaTokenizerFast, LlamaForCausalLM
6
  from accelerate import Accelerator
7
  import torch