lsb commited on
Commit
4380df2
1 Parent(s): aef61e4

segformer fp32 for now

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ seg_model_size = 0
27
  seg_feature_extractor = SegformerFeatureExtractor.from_pretrained(f"nvidia/segformer-b{seg_model_size}-finetuned-cityscapes-{seg_model_img_size}-{seg_model_img_size}")
28
  seg_model = SegformerForSemanticSegmentation.from_pretrained(
29
  f"nvidia/segformer-b{seg_model_size}-finetuned-cityscapes-{seg_model_img_size}-{seg_model_img_size}"
30
- ).to(preferred_device).to(preferred_dtype)
31
 
32
  inpainting_pipeline = StableDiffusionInpaintPipeline.from_pretrained(
33
  "SimianLuo/LCM_Dreamshaper_v7",
 
27
  seg_feature_extractor = SegformerFeatureExtractor.from_pretrained(f"nvidia/segformer-b{seg_model_size}-finetuned-cityscapes-{seg_model_img_size}-{seg_model_img_size}")
28
  seg_model = SegformerForSemanticSegmentation.from_pretrained(
29
  f"nvidia/segformer-b{seg_model_size}-finetuned-cityscapes-{seg_model_img_size}-{seg_model_img_size}"
30
+ ).to(preferred_device) #.to(preferred_dtype)
31
 
32
  inpainting_pipeline = StableDiffusionInpaintPipeline.from_pretrained(
33
  "SimianLuo/LCM_Dreamshaper_v7",