Futuretop commited on
Commit
c9314ca
·
verified ·
1 Parent(s): a184af3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -76,14 +76,13 @@ florence_processor = AutoProcessor.from_pretrained(LOCAL_FLORENCE, trust_remote_
76
  diffusers.StableDiffusion3Pipeline = StableDiffusionPipeline
77
 
78
  model_repo = "tensorart/stable-diffusion-3.5-large-TurboX"
79
- pipe = DiffusionPipeline.from_pretrained(
80
- LOCAL_TURBOX,
 
81
  trust_remote_code=True,
82
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
83
- local_files_only=True,
84
- low_cpu_mem_usage=False,
85
- device_map=None
86
  )
 
87
  pipe.scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(model_repo, subfolder="scheduler", local_files_only=True, trust_remote_code = True, shift=5)
88
  pipe = pipe.to(device)
89
 
 
76
  diffusers.StableDiffusion3Pipeline = StableDiffusionPipeline
77
 
78
  model_repo = "tensorart/stable-diffusion-3.5-large-TurboX"
79
+
80
+ pipe = diffusers.DiffusionPipeline.from_pretrained(
81
+ "tensorart/stable-diffusion-3.5-large-TurboX",
82
  trust_remote_code=True,
83
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
 
 
 
84
  )
85
+
86
  pipe.scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(model_repo, subfolder="scheduler", local_files_only=True, trust_remote_code = True, shift=5)
87
  pipe = pipe.to(device)
88