segformer fp32 for now
Browse files
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)
|
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",
|