xingxm commited on
Commit
0d8e2bd
·
1 Parent(s): 850da3a

fix(dtype): fix undefined variable

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. svgdreamer/painter/VPSD_pipeline.py +2 -2
README.md CHANGED
@@ -114,7 +114,7 @@ python svgdreamer.py x=iconography-s1 skip_sive=False "prompt='an astronaut walk
114
  **Script:**
115
 
116
  ```shell
117
- python svgdreamer.py x=iconography "prompt='Sydney opera house. oil painting. by Van Gogh'" result_path='./logs/SydneyOperaHouse-OilPainting'
118
  ```
119
 
120
  #### ✍️ Painting style
 
114
  **Script:**
115
 
116
  ```shell
117
+ python svgdreamer.py x=iconography "prompt='Sydney opera house. oil painting. by Van Gogh'" result_path='./logs/SydneyOperaHouse-OilPainting' state.mprec='fp16'
118
  ```
119
 
120
  #### ✍️ Painting style
svgdreamer/painter/VPSD_pipeline.py CHANGED
@@ -26,7 +26,7 @@ class VectorizedParticleSDSPipeline(torch.nn.Module):
26
  diffuser_cfg: DictConfig,
27
  guidance_cfg: DictConfig,
28
  device: torch.device,
29
- dtype):
30
  super().__init__()
31
  self.device = device
32
  self.dtype = torch.float16 if precision == 'fp16' else torch.float32
@@ -35,7 +35,7 @@ class VectorizedParticleSDSPipeline(torch.nn.Module):
35
 
36
  pipe_kwargs = {
37
  "device": self.device,
38
- "torch_dtype": torch.float16 if dtype == 'fp16' else torch.float32,
39
  "local_files_only": not diffuser_cfg.download,
40
  "force_download": diffuser_cfg.force_download,
41
  "resume_download": diffuser_cfg.resume_download,
 
26
  diffuser_cfg: DictConfig,
27
  guidance_cfg: DictConfig,
28
  device: torch.device,
29
+ precision: str):
30
  super().__init__()
31
  self.device = device
32
  self.dtype = torch.float16 if precision == 'fp16' else torch.float32
 
35
 
36
  pipe_kwargs = {
37
  "device": self.device,
38
+ "torch_dtype": torch.float16 if precision == 'fp16' else torch.float32,
39
  "local_files_only": not diffuser_cfg.download,
40
  "force_download": diffuser_cfg.force_download,
41
  "resume_download": diffuser_cfg.resume_download,