tow252 commited on
Commit
003ec88
1 Parent(s): adbcafc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -25,8 +25,6 @@ IPmodel_path = "runwayml/stable-diffusion-inpainting"
25
 
26
  IPpipe = StableDiffusionInpaintPipeline.from_pretrained(
27
  IPmodel_path,
28
- revision="fp16",
29
- torch_dtype=torch.float16,
30
  use_auth_token= st.secrets["AUTH_TOKEN"]
31
  ).to(device)
32
 
@@ -34,7 +32,7 @@ trans_tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600
34
  trans_model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M")
35
 
36
 
37
- SDpipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", revision="fp16", torch_dtype=torch.float16, use_auth_token=st.secrets["AUTH_TOKEN"]).to(device)
38
 
39
 
40
  def create_mask(image, prompt):
 
25
 
26
  IPpipe = StableDiffusionInpaintPipeline.from_pretrained(
27
  IPmodel_path,
 
 
28
  use_auth_token= st.secrets["AUTH_TOKEN"]
29
  ).to(device)
30
 
 
32
  trans_model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M")
33
 
34
 
35
+ SDpipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", use_auth_token=st.secrets["AUTH_TOKEN"]).to(device)
36
 
37
 
38
  def create_mask(image, prompt):